Home
last modified time | relevance | path

Searched refs:HloGetTupleElementInstruction (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc1695 HloGetTupleElementInstruction::HloGetTupleElementInstruction( in HloGetTupleElementInstruction() function in xla::HloGetTupleElementInstruction
1701 HloInstructionProto HloGetTupleElementInstruction::ToProto() const { in ToProto()
1707 std::vector<string> HloGetTupleElementInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()
1712 bool HloGetTupleElementInstruction::IdenticalSlowPath( in IdenticalSlowPath()
1717 static_cast<const HloGetTupleElementInstruction&>(other); in IdenticalSlowPath()
1722 HloGetTupleElementInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()
1726 return absl::make_unique<HloGetTupleElementInstruction>( in CloneWithNewOperandsImpl()
Dhlo_instructions.h908 class HloGetTupleElementInstruction : public HloInstruction {
910 explicit HloGetTupleElementInstruction(const Shape& shape,
Dhlo_instruction.cc661 return absl::make_unique<HloGetTupleElementInstruction>(shape, operand, in CreateGetTupleElement()
3385 return Cast<HloGetTupleElementInstruction>(this)->tuple_index(); in tuple_index()