Home
last modified time | relevance | path

Searched refs:hlo_metadatas_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_program_group.cc110 hlo_metadatas_ = hlo_metadatas; in Initialize()
158 hlo_metadatas_(std::move(other.hlo_metadatas_)) { in TpuProgramGroup()
164 hlo_metadatas_.resize(hlo_metadatas.size()); in set_hlo_metadatas()
166 hlo_metadatas_[i] = hlo_metadatas[i]; in set_hlo_metadatas()
182 hlo_metadatas_ptrs_.reserve(hlo_metadatas_.size()); in RefreshHloMetadatasPtrs()
183 for (const auto& hlo_metadata_internal_ : hlo_metadatas_) { in RefreshHloMetadatasPtrs()
Dtpu_program_group.h181 std::vector<xla::HloProto> hlo_metadatas_; // Owned. variable