Searched refs:XLA_TpuProgram (Results 1 – 7 of 7) sorted by relevance
34 typedef struct XLA_TpuProgram XLA_TpuProgram; typedef96 XLA_TpuProgram** tpu_programs[], size_t* count, TF_Status* status);106 XLA_TpuProgram** tpu_programs[], size_t* count, TF_Status* status);158 const XLA_TpuProgram* program;319 TFTPU_CAPI_EXPORT XLA_TpuProgram* TpuProgram_New();322 TFTPU_CAPI_EXPORT void TpuProgram_Free(XLA_TpuProgram* tpu_program);325 TFTPU_CAPI_EXPORT XLA_TpuProgram** TpuProgram_NewArray(size_t count);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);[all …]
50 const XLA_TpuProgram* tpu_program);
422 const XLA_TpuProgram* tpu_program) { in TPUExecute()
37 const XLA_TpuProgram* xla_tpu_program) { in ConstructExecutableInfo()53 const XLA_TpuProgram* xla_tpu_program) { in ConstructHostTransferInfo()69 const XLA_TpuProgram* xla_tpu_program) { in ConstructHloMetadata()85 absl::Span<XLA_TpuProgram* const> xla_tpu_programs) { in Initialize()97 const XLA_TpuProgram* xla_tpu_program = tpu_programs_[i]; in Initialize()115 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in has_sharding_program()127 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in program_size()135 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in LogProgramMemorySummary()142 for (XLA_TpuProgram* tpu_program : tpu_programs_) { in UnloadAndDestroyPrograms()213 const std::vector<XLA_TpuProgram*>& TpuProgramGroup::tpu_programs() const { in tpu_programs()[all …]
105 void Initialize(absl::Span<XLA_TpuProgram* const> xla_tpu_programs);128 const std::vector<XLA_TpuProgram*>& tpu_programs() const;129 std::vector<XLA_TpuProgram*> tpu_programs(TpuProgramShardingType type) const;130 const XLA_TpuProgram* tpu_program(int index) const;131 void set_tpu_programs(absl::Span<XLA_TpuProgram* const> tpu_programs);161 const XLA_TpuProgram* tpu_program);163 const XLA_TpuProgram* tpu_program);164 xla::HloProto ConstructHloMetadata(const XLA_TpuProgram* tpu_program);172 std::vector<XLA_TpuProgram*> tpu_programs_; // Not owned.
42 explicit TpuExecutable(const XLA_TpuProgram* core_program,47 const XLA_TpuProgram* core_program() const { return core_program_; } in core_program()63 const XLA_TpuProgram* const core_program_;
30 TpuExecutable::TpuExecutable(const XLA_TpuProgram* core_program, in TpuExecutable()