Searched refs:TpuMeshCommonState (Results 1 – 2 of 2) sorted by relevance
27 class TpuMeshCommonState; variable51 tensorflow::TpuMeshCommonState* mesh_common_state() const { in mesh_common_state()52 return static_cast<tensorflow::TpuMeshCommonState*>( in mesh_common_state()
29 class TpuMeshCommonState; variable230 tensorflow::TpuMeshCommonState* tpu_mesh_common_state;