Searched defs:sparse_end (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/XNNPACK/bench/ |
| H A D | f16-spmm.cc | 53 const size_t sparse_end = std::min(size_t(float(b.size()) * sparsity), b.size()); in f16_spmm() local
|
| H A D | f32-spmm.cc | 50 const size_t sparse_end = std::min(size_t(float(b.size()) * sparsity), b.size()); in f32_spmm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 1966 ArrayRef<int64_t> sparse_end, ArrayRef<int64_t> sparse_strides, in CalculateSlicedShapeFromSparseIndices()
2014 SmallVector<int64_t, 4> sparse_begin, sparse_end, sparse_strides; in GetSlicedBoundRanges() local
2186 SmallVector<int64_t, 4> sparse_begin, sparse_end, sparse_strides; in GetSlicedShapeAndBoundRanges() local
|