Searched defs:hlo_metadatas (Results 1 – 3 of 3) sorted by relevance
95 std::vector<xla::HloProto> hlo_metadatas(tpu_programs_.size()); in Initialize() local163 absl::Span<const xla::HloProto> hlo_metadatas) { in set_hlo_metadatas()171 absl::Span<const xla::HloProto* const> TpuProgramGroup::hlo_metadatas() const { in hlo_metadatas() function in tensorflow::tpu::TpuProgramGroup
367 absl::Span<const xla::HloProto* const>* hlo_metadatas, in CompileIfKeyAbsent()406 absl::Span<const xla::HloProto* const>* hlo_metadatas, in CompileIfKeyAbsentHelper()
691 absl::Span<const xla::HloProto* const> hlo_metadatas; in ComputeInternal() local