Searched refs:slice_dims (Results 1 – 4 of 4) sorted by relevance
674 absl::InlinedVector<int64, 4> slice_dims; in Compile() local682 slice_dims.push_back(end[i] - begin[i]); in Compile()688 slice_dims.push_back(begin[i] - end[i]); in Compile()696 rhs = xla::Reshape(rhs, slice_dims); in Compile()
405 auto slice_dims = value_shape.dim_sizes(); in Compile() local406 slice_dims[0] = 1LL; in Compile()426 ta = DynamicAddSlice(b, ta, slice, slice_dims, start_indices, dtype_); in Compile()
102 int64 slice_dims = max_indices.size(); in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims() local103 if (slice_dims > 1) { in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()106 {slice_dims}), in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
2322 std::vector<int64> slice_dims; in HandleDynamicUpdateSlice() local2327 slice_dims.push_back(i); in HandleDynamicUpdateSlice()2369 slice_dims); in HandleDynamicUpdateSlice()