Searched defs:root_instruction (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_computation.h66 // Build and return an HloComputation. The parameter root_instruction
68 // root_instruction is nullptr then use the last added instruction as the
71 HloInstruction* root_instruction = nullptr);
125 HloInstruction* root_instruction() const { return root_instruction_; } function in class:xla::HloComputation
349 HloInstruction* root_instruction, HloInstruction* fusion_instruction);
H A Dhlo_computation.cc48 HloInstruction* root_instruction) {
55 // If root_instruction is not specified use the last added instruction.
57 root_instruction ? root_instruction : last_added_instruction_;
66 HloInstruction* root_instruction, HloInstruction* fusion_instruction)
68 root_instruction_(root_instruction),
193 TF_RET_CHECK(root_instruction() != instruction);
206 item == root_instruction() || !IsRemovable(item) ||
225 TF_RET_CHECK(root_instruction() != instruction)
401 proto.set_root_name(root_instruction()
47 Build( HloInstruction* root_instruction) argument
63 HloComputation( const string& name, int parameter_count, std::vector<std::unique_ptr<HloInstruction>>* instructions, HloInstruction* root_instruction, HloInstruction* fusion_instruction) argument
[all...]

Completed in 283 milliseconds