Searched refs:sparse_begin (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1825 MutableArrayRef<int64_t> input_shape, ArrayRef<int64_t> sparse_begin, in CalculateSlicedShapeFromSparseIndices() argument
1831 int64_t num_sparse_indices = sparse_begin.size(); in CalculateSlicedShapeFromSparseIndices()
1834 sparse_begin, sparse_end, sparse_strides}; in CalculateSlicedShapeFromSparseIndices()
1874 SmallVector<int64_t, 4> sparse_begin, sparse_end, sparse_strides; in GetSlicedBoundRanges() local
1877 sparse_begin.push_back(index.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
2048 sparse_begin.push_back(index.getSExtValue()); in GetSlicedShapeAndBoundRanges()
2055 *input_shape, sparse_begin, sparse_end, sparse_strides, begin_mask(), in GetSlicedShapeAndBoundRanges()
|