Searched defs:hlo_metadata (Results 1 – 4 of 4) sorted by relevance
75 xla::HloProto hlo_metadata; in ConstructHloMetadata() local175 const xla::HloProto* TpuProgramGroup::hlo_metadata(int index) const { in hlo_metadata() function in tensorflow::tpu::TpuProgramGroup
800 for (const xla::HloProto* hlo_metadata : hlo_metadatas) { in ComputeInternal() local
196 absl::Span<const xla::HloProto* const> hlo_metadata; in Compute() local
415 const xla::HloProto& hlo_metadata, in TPUExecute()