Home
last modified time | relevance | path

Searched refs:TpuSerializedProto (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h59 typedef struct TpuSerializedProto { struct
62 } TpuSerializedProto; argument
167 TpuSerializedProto device_assignment;
282 TpuSerializedProto debug_options;
284 TpuSerializedProto static_device_assignment;
297 TpuSerializedProto proto;
302 TpuSerializedProto proto;
Dproto_helper.h26 void StreamExecutor_Tpu_FreeSerializedProto(const TpuSerializedProto* proto);
33 using SerializedProto = TpuSerializedProto;
Dproto_helper.cc20 void StreamExecutor_Tpu_FreeSerializedProto(const TpuSerializedProto* proto) { in StreamExecutor_Tpu_FreeSerializedProto()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h26 typedef struct TpuSerializedProto TpuSerializedProto; typedef
95 TpuSerializedProto compilation_request, const XLA_TpuMeshState* mesh_state,
105 TpuSerializedProto xrt_computation, const XLA_TpuMeshState* mesh_state,
345 const XLA_TpuProgram* tpu_program, TpuSerializedProto* executable_info,
350 const XLA_TpuProgram* tpu_program, TpuSerializedProto* host_transfer_info,
355 const XLA_TpuProgram* tpu_program, TpuSerializedProto* hlo_metadata,
383 TpuSerializedProto get_tpu_program_response, XLA_TpuProgram* tpu_program,
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_program_group.cc39 TpuSerializedProto serialized_executable_info = {}; in ConstructExecutableInfo()
55 TpuSerializedProto serialized_host_transfer_info = {}; in ConstructHostTransferInfo()
71 TpuSerializedProto serialized_hlo_metadata = {}; in ConstructHloMetadata()
330 const std::vector<TpuSerializedProto>& rpc_response_protos) { in DeserializeFromRpcResponseProtos()
Dtpu_compilation_cache_rpc_support.cc47 TpuSerializedProto serialized_response_proto = in DeserializeRpcResponseToCacheEntry()
Dtpu_program_group.h142 const std::vector<TpuSerializedProto>& rpc_response_protos);