Searched refs:dynamic_slice_sizes_ (Results 1 – 2 of 2) sorted by relevance
1319 return dynamic_slice_sizes_[dimension]; in slice_sizes()1322 return dynamic_slice_sizes_; in dynamic_slice_sizes()1343 std::vector<int64> dynamic_slice_sizes_; variable
2237 dynamic_slice_sizes_(slice_sizes.begin(), slice_sizes.end()) { in HloDynamicSliceInstruction()2247 dynamic_slice_sizes_(slice_sizes.begin(), slice_sizes.end()) { in HloDynamicSliceInstruction()2276 for (int64 slice_size : dynamic_slice_sizes_) { in ToProto()2302 shape, new_operands[0], new_operands[1], dynamic_slice_sizes_); in CloneWithNewOperandsImpl()2305 shape, new_operands[0], new_operands.subspan(1), dynamic_slice_sizes_); in CloneWithNewOperandsImpl()