Searched refs:HloGetDimensionSizeInstruction (Results 1 – 3 of 3) sorted by relevance
3076 HloGetDimensionSizeInstruction::HloGetDimensionSizeInstruction( in HloGetDimensionSizeInstruction() function in xla::HloGetDimensionSizeInstruction3083 HloInstructionProto HloGetDimensionSizeInstruction::ToProto() const { in ToProto()3089 std::vector<string> HloGetDimensionSizeInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()3094 bool HloGetDimensionSizeInstruction::IdenticalSlowPath( in IdenticalSlowPath()3099 static_cast<const HloGetDimensionSizeInstruction&>(other); in IdenticalSlowPath()3104 HloGetDimensionSizeInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()3110 return absl::make_unique<HloGetDimensionSizeInstruction>( in CloneWithNewOperandsImpl()
1852 class HloGetDimensionSizeInstruction : public HloInstruction {1854 explicit HloGetDimensionSizeInstruction(const Shape& shape,
1372 return absl::make_unique<HloGetDimensionSizeInstruction>(shape, operand, in CreateGetDimensionSize()3918 return Cast<HloGetDimensionSizeInstruction>(this)->dimension(); in dimension()