Home
last modified time | relevance | path

Searched refs:computation_mesh (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmove_compilation_to_host.cc62 const Mesh& computation_mesh = mesh_or->value(); in IdentifyAndValidateMeshComputations() local
63 if (function_map->count(computation_mesh)) in IdentifyAndValidateMeshComputations()
68 (*function_map)[computation_mesh] = dtensor_function; in IdentifyAndValidateMeshComputations()