Home
last modified time | relevance | path

Searched defs:mesh_state (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_compile_ops.cc92 tensorflow::tpu::TpuMeshStateInterface* mesh_state, int num_replicas, in CompilationCacheKey()
158 tensorflow::tpu::TpuMeshStateInterface* mesh_state; in Compute() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_impl.cc30 const XLA_TpuMeshState* mesh_state, in Compile()
Dtpu_compile_op_common.cc147 const TpuMeshStateInterface* mesh_state, in CompileLocallyAndFillHostCache()
193 TpuMeshStateInterface* mesh_state; in ComputeInternal() local
Dtpu_compile_op_common.h130 const TpuMeshStateInterface* mesh_state, in LookupPersistentCompilationCacheAndFillCaches()
Dtpu_program_group.cc269 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
303 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
Dtpu_op_util.cc109 const TpuMeshStateInterface& mesh_state) { in CreateCompilationCacheKey()
Dtpu_configuration_ops.cc206 tpu::TpuMeshStateInterface* mesh_state; in Compute() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h81 const XLA_TpuMeshState* mesh_state; member