Searched refs:sparse_strides (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1826 ArrayRef<int64_t> sparse_end, ArrayRef<int64_t> sparse_strides, in CalculateSlicedShapeFromSparseIndices() argument
1834 sparse_begin, sparse_end, sparse_strides}; in CalculateSlicedShapeFromSparseIndices()
1874 SmallVector<int64_t, 4> sparse_begin, sparse_end, sparse_strides; in GetSlicedBoundRanges() local
1881 sparse_strides.push_back(stride.getSExtValue()); in GetSlicedBoundRanges()
1884 input_shape, sparse_begin, sparse_end, sparse_strides, begin_mask(), in GetSlicedBoundRanges()
2045 SmallVector<int64_t, 4> sparse_begin, sparse_end, sparse_strides; in GetSlicedShapeAndBoundRanges() local
2052 sparse_strides.push_back(stride.getSExtValue()); in GetSlicedShapeAndBoundRanges()
2055 *input_shape, sparse_begin, sparse_end, sparse_strides, begin_mask(), in GetSlicedShapeAndBoundRanges()
|