Searched defs:mesh_state (Results 1 – 8 of 8) sorted by relevance
92 tensorflow::tpu::TpuMeshStateInterface* mesh_state, int num_replicas, in CompilationCacheKey()158 tensorflow::tpu::TpuMeshStateInterface* mesh_state; in Compute() local
30 const XLA_TpuMeshState* mesh_state, in Compile()
147 const TpuMeshStateInterface* mesh_state, in CompileLocallyAndFillHostCache()193 TpuMeshStateInterface* mesh_state; in ComputeInternal() local
130 const TpuMeshStateInterface* mesh_state, in LookupPersistentCompilationCacheAndFillCaches()
269 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()303 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
109 const TpuMeshStateInterface& mesh_state) { in CreateCompilationCacheKey()
206 tpu::TpuMeshStateInterface* mesh_state; in Compute() local
81 const XLA_TpuMeshState* mesh_state; member