Searched refs:XLA_TpuMeshState (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_ops_c_api.h | 54 typedef struct XLA_TpuMeshState XLA_TpuMeshState; typedef
75 const XLA_TpuMeshState* mesh_state;
95 TpuSerializedProto compilation_request, const XLA_TpuMeshState* mesh_state,
105 TpuSerializedProto xrt_computation, const XLA_TpuMeshState* mesh_state,
144 TFTPU_CAPI_EXPORT XLA_TpuMeshState* TpuMeshState_Create();
148 TFTPU_CAPI_EXPORT void TpuMeshState_Free(XLA_TpuMeshState* mesh_state);
152 XLA_TpuMeshState* mesh_state);
397 const XLA_TpuMeshState* mesh_state, TpuCoreTypeEnum tpu_core_type);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_mesh_state_interface.h | 35 explicit TpuMeshStateInterface(XLA_TpuMeshState* handle) in TpuMeshStateInterface()
49 const XLA_TpuMeshState* data() const { return mesh_state_; } in data()
72 XLA_TpuMeshState* mesh_state_;
|
| D | tpu_program_group.h | 95 const XLA_TpuMeshState* mesh_state,
101 const XLA_TpuMeshState* mesh_state,
|
| D | tpu_compile_op_impl.h | 56 const XLA_TpuMeshState* mesh_state,
|
| D | tpu_compile_op_impl.cc | 30 const XLA_TpuMeshState* mesh_state, in Compile()
|
| D | tpu_program_group.cc | 248 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
282 const XLA_TpuMeshState* mesh_state, in CompileAndBuild()
|
| D | tpu_compile_op_common.h | 103 const XLA_TpuMeshState* mesh_state,
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 75 Status Compile(const XLA_TpuMeshState* xla_mesh_state,
83 const XLA_TpuMeshState* xla_mesh_state, in Compile()
|