Searched refs:tpu_programs (Results 1 – 4 of 4) sorted by relevance
213 const std::vector<XLA_TpuProgram*>& TpuProgramGroup::tpu_programs() const { in tpu_programs() function in tensorflow::tpu::TpuProgramGroup224 absl::Span<XLA_TpuProgram* const> tpu_programs) { in set_tpu_programs() argument225 tpu_programs_.resize(tpu_programs.size()); in set_tpu_programs()226 for (size_t i = 0; i < tpu_programs.size(); ++i) { in set_tpu_programs()227 tpu_programs_[i] = tpu_programs[i]; in set_tpu_programs()315 std::vector<XLA_TpuProgram*> TpuProgramGroup::tpu_programs( in tpu_programs() function in tensorflow::tpu::TpuProgramGroup317 std::vector<XLA_TpuProgram*> tpu_programs; in tpu_programs() local318 tpu_programs.reserve(tpu_programs_.size()); in tpu_programs()321 tpu_programs.push_back(OpsApiFn()->TpuProgram_GetTpuProgramFn( in tpu_programs()323 CHECK_NE(tpu_programs[i], nullptr); in tpu_programs()[all …]
128 const std::vector<XLA_TpuProgram*>& tpu_programs() const;129 std::vector<XLA_TpuProgram*> tpu_programs(TpuProgramShardingType type) const;131 void set_tpu_programs(absl::Span<XLA_TpuProgram* const> tpu_programs);
120 tpu_program_group.tpu_programs(TpuProgramShardingType::kSharding)); in InitializeEntry()128 tpu_program_group.tpu_programs(TpuProgramShardingType::kUnsharding)); in InitializeEntry()
96 XLA_TpuProgram** tpu_programs[], size_t* count, TF_Status* status);106 XLA_TpuProgram** tpu_programs[], size_t* count, TF_Status* status);