Searched refs:dynamic_slice_sizes_ (Results 1 – 2 of 2) sorted by relevance
1612 return dynamic_slice_sizes_[dimension]; in slice_sizes()1615 return dynamic_slice_sizes_; in dynamic_slice_sizes()1636 std::vector<int64> dynamic_slice_sizes_; variable
2650 dynamic_slice_sizes_(slice_sizes.begin(), slice_sizes.end()) { in HloDynamicSliceInstruction()2660 dynamic_slice_sizes_(slice_sizes.begin(), slice_sizes.end()) { in HloDynamicSliceInstruction()2689 for (int64 slice_size : dynamic_slice_sizes_) { in ToProto()2716 shape, new_operands[0], new_operands[1], dynamic_slice_sizes_); in CloneWithNewOperandsImpl()2719 shape, new_operands[0], new_operands.subspan(1), dynamic_slice_sizes_); in CloneWithNewOperandsImpl()