Skip to main content

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?