Searched refs:dtensor_function (Results 1 – 1 of 1) sorted by relevance
55 for (auto dtensor_function : in IdentifyAndValidateMeshComputations() local57 auto mesh_or = ExtractDeviceMeshFromOp(dtensor_function); in IdentifyAndValidateMeshComputations()59 return dtensor_function.emitOpError( in IdentifyAndValidateMeshComputations()64 return dtensor_function.emitOpError( in IdentifyAndValidateMeshComputations()68 (*function_map)[computation_mesh] = dtensor_function; in IdentifyAndValidateMeshComputations()