Searched defs:mesh_state (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_embedding_ops.cc | 59 tensorflow::tpu::TpuMeshStateInterface* mesh_state; in Compile() local
141 tensorflow::tpu::TpuMeshStateInterface* mesh_state; in Compile() local
213 tensorflow::tpu::TpuMeshStateInterface* mesh_state; in Compile() local
|
| H A D | tpu_compile_op_common.cc | 165 const TpuMeshStateInterface* mesh_state, in CompileLocallyAndFillHostCache()
180 const TpuMeshStateInterface* mesh_state, in CompileLocallyAndFillHostCacheInternal()
227 TpuMeshStateInterface* mesh_state; in ComputeInternal() local
|
| H A D | tpu_program_group.cc | 262 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
296 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
|
| H A D | tpu_compile_op_impl.cc | 30 const XLA_TpuMeshState* mesh_state, in Compile()
|
| H A D | tpu_compile_op_common.h | 140 const TpuMeshStateInterface* mesh_state, in LookupPersistentCompilationCacheAndFillCaches()
|
| H A D | tpu_op_util.cc | 110 const TpuMeshStateInterface& mesh_state, uint64_t session_id, in CreateCompilationCacheKey()
|
| H A D | tpu_embedding_configuration_ops.cc | 359 tensorflow::tpu::TpuMeshStateInterface* mesh_state; in Compute() local
|
| H A D | tpu_configuration_ops.cc | 212 tpu::TpuMeshStateInterface* mesh_state; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | tpu_compile_ops.cc | 93 tensorflow::tpu::TpuMeshStateInterface* mesh_state, int num_replicas, in CompilationCacheKey()
159 tensorflow::tpu::TpuMeshStateInterface* mesh_state; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_ops_c_api.h | 91 const XLA_TpuMeshState* mesh_state; member
|