Mojo function
rebuild_mix_precision_static_tensor_specs_with_output_lambda
rebuild_mix_precision_static_tensor_specs_with_output_lambda[func_type: AnyTrivialRegType, //, src_rank: Int, src_shape: DimList, src_type: DType](spec: StaticTensorSpec[dtype, rank], out_lambda: func_type) -> StaticTensorSpec[src_type, src_rank]
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!