Home
last modified time | relevance | path

Searched refs:sparse_end (Results 1 – 3 of 3) sorted by relevance

/external/XNNPACK/bench/
Df16-spmm.cc54 const size_t sparse_end = std::min(size_t(float(b.size()) * sparsity), b.size()); in SpMMBenchmark() local
55 const size_t num_nonzeroes = nr * (b.size() - sparse_end); in SpMMBenchmark()
74 std::fill(b.begin(), b.begin() + sparse_end, 0); in SpMMBenchmark()
75 std::generate(b.begin() + sparse_end, b.end(), std::ref(f16rng)); in SpMMBenchmark()
Df32-spmm.cc47 const size_t sparse_end = std::min(size_t(float(b.size()) * sparsity), b.size()); in SpMMBenchmark() local
48 const size_t num_nonzeroes = nr * (b.size() - sparse_end); in SpMMBenchmark()
67 std::fill(b.begin(), b.begin() + sparse_end, 0.0f); in SpMMBenchmark()
68 std::generate(b.begin() + sparse_end, b.end(), std::ref(f32rng)); in SpMMBenchmark()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1826 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
1879 sparse_end.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
2050 sparse_end.push_back(index.getSExtValue()); in GetSlicedShapeAndBoundRanges()
2055 *input_shape, sparse_begin, sparse_end, sparse_strides, begin_mask(), in GetSlicedShapeAndBoundRanges()