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()
132 const TpuMeshStateInterface* mesh_state, in LookupPersistentCompilationCacheAndFillCaches()
248 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()282 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
109 const TpuMeshStateInterface& mesh_state) { in CreateCompilationCacheKey()
593 const TpuMeshStateInterface* mesh_state, in CompileLocallyAndFillHostCache()635 TpuMeshStateInterface* mesh_state; in ComputeInternal() local
190 tpu::TpuMeshStateInterface* mesh_state; in Compute() local
75 const XLA_TpuMeshState* mesh_state; member