Searched refs:hlo_metadatas (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_program_group.cc | 95 std::vector<xla::HloProto> hlo_metadatas(tpu_programs_.size()); in Initialize() local
104 hlo_metadatas[i] = ConstructHloMetadata(xla_tpu_program); in Initialize()
110 hlo_metadatas_ = hlo_metadatas; in Initialize()
163 absl::Span<const xla::HloProto> hlo_metadatas) { in set_hlo_metadatas() argument
164 hlo_metadatas_.resize(hlo_metadatas.size()); in set_hlo_metadatas()
165 for (size_t i = 0; i < hlo_metadatas.size(); ++i) { in set_hlo_metadatas()
166 hlo_metadatas_[i] = hlo_metadatas[i]; in set_hlo_metadatas()
171 absl::Span<const xla::HloProto* const> TpuProgramGroup::hlo_metadatas() const { in hlo_metadatas() function in tensorflow::tpu::TpuProgramGroup
|
| D | tpu_program_group.h | 136 void set_hlo_metadatas(absl::Span<const xla::HloProto> hlo_metadatas);
138 absl::Span<const xla::HloProto* const> hlo_metadatas() const override;
|
| D | tpu_compilation_cache_interface.cc | 367 absl::Span<const xla::HloProto* const>* hlo_metadatas, in CompileIfKeyAbsent() argument
372 sharding_key, may_modify_variables, &removed_entries, hlo_metadatas, in CompileIfKeyAbsent()
406 absl::Span<const xla::HloProto* const>* hlo_metadatas, in CompileIfKeyAbsentHelper() argument
505 *hlo_metadatas = entry->tpu_program_group->hlo_metadatas(); in CompileIfKeyAbsentHelper()
|
| D | tpu_program_group_interface.h | 60 virtual absl::Span<const xla::HloProto* const> hlo_metadatas() const = 0;
|
| D | tpu_compilation_cache_interface.h | 114 absl::Span<const xla::HloProto* const>* hlo_metadatas,
204 absl::Span<const xla::HloProto* const>* hlo_metadatas,
|
| D | tpu_compile_op_common.cc | 691 absl::Span<const xla::HloProto* const> hlo_metadatas; in ComputeInternal() local
694 &may_modify_variables, &hlo_metadatas, in ComputeInternal()
800 for (const xla::HloProto* hlo_metadata : hlo_metadatas) { in ComputeInternal()
|