Searched refs:HloSliceInstruction (Results 1 – 3 of 3) sorted by relevance
1387 HloSliceInstruction::HloSliceInstruction( in HloSliceInstruction() function in xla::HloSliceInstruction1404 HloInstructionProto HloSliceInstruction::ToProto() const { in ToProto()1415 std::vector<std::string> HloSliceInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1429 bool HloSliceInstruction::IdenticalSlowPath( in IdenticalSlowPath()1433 const auto& other_slice = static_cast<const HloSliceInstruction&>(other); in IdenticalSlowPath()1439 std::unique_ptr<HloInstruction> HloSliceInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1443 return std::make_unique<HloSliceInstruction>( in CloneWithNewOperandsImpl()1492 const auto& other_slice = static_cast<const HloSliceInstruction&>(other); in IdenticalSlowPath()
1509 return std::make_unique<HloSliceInstruction>(shape, operand, start_indices, in CreateSlice()4499 return Cast<HloSliceInstruction>(this)->slice_starts(dimension); in slice_starts()4503 return Cast<HloSliceInstruction>(this)->slice_starts(); in slice_starts()4507 return Cast<HloSliceInstruction>(this)->mutable_slice_starts(); in mutable_slice_starts()4511 return Cast<HloSliceInstruction>(this)->slice_limits(dimension); in slice_limits()4515 return Cast<HloSliceInstruction>(this)->slice_limits(); in slice_limits()4519 return Cast<HloSliceInstruction>(this)->mutable_slice_limits(); in mutable_slice_limits()4523 return Cast<HloSliceInstruction>(this)->slice_strides(dimension); in slice_strides()4527 return Cast<HloSliceInstruction>(this)->slice_strides(); in slice_strides()4531 return Cast<HloSliceInstruction>(this)->mutable_slice_strides(); in mutable_slice_strides()
1045 class HloSliceInstruction : public HloInstruction {1047 explicit HloSliceInstruction(const Shape& shape, HloInstruction* operand,