Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h322 TFTPU_CAPI_EXPORT void TpuProgram_Free(XLA_TpuProgram* tpu_program);
328 TFTPU_CAPI_EXPORT void TpuProgram_FreeArray(XLA_TpuProgram* tpu_program[]);
332 TFTPU_CAPI_EXPORT void TpuProgram_UnloadAndDestroy(XLA_TpuProgram* tpu_program,
337 TpuProgram_GetProgramSize(const XLA_TpuProgram* tpu_program);
341 const XLA_TpuProgram* tpu_program);
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,
360 const XLA_TpuProgram* tpu_program, bool* may_modify_variables);
364 const XLA_TpuProgram* tpu_program);
[all …]
Dtpu_execute.h50 const XLA_TpuProgram* tpu_program);
Dtpu_execute.cc422 const XLA_TpuProgram* tpu_program) { in TPUExecute() argument
425 TF_RET_CHECK(tpu_program != nullptr); in TPUExecute()
482 tpu_program, std::move(module), /*host_command_handler=*/handler); in TPUExecute()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_program_group.cc115 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in has_sharding_program() local
116 if (!OpsApiFn()->TpuProgram_HasShardingFn(tpu_program)) { in has_sharding_program()
127 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in program_size() local
128 total_size += OpsApiFn()->TpuProgram_GetProgramSizeFn(tpu_program); in program_size()
135 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in LogProgramMemorySummary() local
136 success &= OpsApiFn()->TpuProgram_LogProgramMemorySummaryFn(tpu_program); in LogProgramMemorySummary()
142 for (XLA_TpuProgram* tpu_program : tpu_programs_) { in UnloadAndDestroyPrograms()
144 OpsApiFn()->TpuProgram_UnloadAndDestroyFn(tpu_program, status.c_status); in UnloadAndDestroyPrograms()
217 const XLA_TpuProgram* TpuProgramGroup::tpu_program(int index) const { in tpu_program() function in tensorflow::tpu::TpuProgramGroup
Dtpu_program_group.h130 const XLA_TpuProgram* tpu_program(int index) const;
161 const XLA_TpuProgram* tpu_program);
163 const XLA_TpuProgram* tpu_program);
164 xla::HloProto ConstructHloMetadata(const XLA_TpuProgram* tpu_program);
Dtpu_execute_op.cc761 tpu_program_group->tpu_program(core_index))); in DoWork()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc222 tpu_program_group->tpu_program(core_index))); in RunExecutable()