Searched refs:HloSliceInstruction (Results 1 – 3 of 3) sorted by relevance
963 HloSliceInstruction::HloSliceInstruction(const Shape& shape, in HloSliceInstruction() function in xla::HloSliceInstruction981 HloInstructionProto HloSliceInstruction::ToProto() const { in ToProto()992 std::vector<string> HloSliceInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1006 bool HloSliceInstruction::IdenticalSlowPath( in IdenticalSlowPath()1010 const auto& other_slice = static_cast<const HloSliceInstruction&>(other); in IdenticalSlowPath()1016 std::unique_ptr<HloInstruction> HloSliceInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1020 return absl::make_unique<HloSliceInstruction>( in CloneWithNewOperandsImpl()1064 const auto& other_slice = static_cast<const HloSliceInstruction&>(other); in IdenticalSlowPath()
988 return absl::make_unique<HloSliceInstruction>(shape, operand, start_indices, in CreateSlice()3248 return Cast<HloSliceInstruction>(this)->slice_starts(dimension); in slice_starts()3252 return Cast<HloSliceInstruction>(this)->slice_starts(); in slice_starts()3256 return Cast<HloSliceInstruction>(this)->slice_limits(dimension); in slice_limits()3260 return Cast<HloSliceInstruction>(this)->slice_limits(); in slice_limits()3264 return Cast<HloSliceInstruction>(this)->slice_strides(dimension); in slice_strides()3268 return Cast<HloSliceInstruction>(this)->slice_strides(); in slice_strides()
608 class HloSliceInstruction : public HloInstruction {610 explicit HloSliceInstruction(const Shape& shape, HloInstruction* operand,