Mojo function
rebuild_mix_precision_static_tensor_specs_with_input_lambda
rebuild_mix_precision_static_tensor_specs_with_input_lambda[func_type: AnyTrivialRegType, //, src_dtype: DType, dst_dtype: DType, rank: Int](spec: StaticTensorSpec[src_dtype, rank], in_lambda: func_type) -> StaticTensorSpec[dst_dtype, rank]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!