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 | 1965 MutableArrayRef<int64_t> input_shape, ArrayRef<int64_t> sparse_begin, in CalculateSlicedShapeFromSparseIndices() argument
1971 int64_t num_sparse_indices = sparse_begin.size(); in CalculateSlicedShapeFromSparseIndices()
1974 sparse_begin, sparse_end, sparse_strides}; in CalculateSlicedShapeFromSparseIndices()
2014 SmallVector<int64_t, 4> sparse_begin, sparse_end, sparse_strides; in GetSlicedBoundRanges() local
2017 sparse_begin.push_back(index.getSExtValue()); in GetSlicedBoundRanges()
2024 input_shape, sparse_begin, sparse_end, sparse_strides, begin_mask(), in GetSlicedBoundRanges()
2186 SmallVector<int64_t, 4> sparse_begin, sparse_end, sparse_strides; in GetSlicedShapeAndBoundRanges() local
2189 sparse_begin.push_back(index.getSExtValue()); in GetSlicedShapeAndBoundRanges()
2196 *input_shape, sparse_begin, sparse_end, sparse_strides, begin_mask(), in GetSlicedShapeAndBoundRanges()
|