Searched refs:hlo_metadata (Results 1 – 9 of 9) sorted by relevance
196 absl::Span<const xla::HloProto* const> hlo_metadata; in Compute() local201 &may_modify_variables, &hlo_metadata, in Compute()215 hlo_metadata[i]->hlo_module().host_program_shape(); in Compute()
132 tpu_program_group->hlo_metadata(core_index) in GetExecutableAliasConfig()217 *tpu_program_group->hlo_metadata(core_index), in RunExecutable()
75 xla::HloProto hlo_metadata; in ConstructHloMetadata() local77 hlo_metadata = in ConstructHloMetadata()81 return hlo_metadata; in ConstructHloMetadata()175 const xla::HloProto* TpuProgramGroup::hlo_metadata(int index) const { in hlo_metadata() function in tensorflow::tpu::TpuProgramGroup
137 const xla::HloProto* hlo_metadata(int index) const;
800 for (const xla::HloProto* hlo_metadata : hlo_metadatas) { in ComputeInternal() local802 *hlo_proto = *hlo_metadata; in ComputeInternal()
757 *tpu_program_group->hlo_metadata(core_index), std::move(input), in DoWork()
43 const xla::HloProto& hlo_metadata,
415 const xla::HloProto& hlo_metadata, in TPUExecute() argument469 hlo_metadata.hlo_module().input_output_alias())); in TPUExecute()472 for (auto& prefetch : hlo_metadata.hlo_module().cross_program_prefetches()) { in TPUExecute()
355 const XLA_TpuProgram* tpu_program, TpuSerializedProto* hlo_metadata,