Home
last modified time | relevance | path

Searched defs:SparseSlice (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_matmul_op.cc107 struct SparseSlice { struct
108 using ConstMatrixMap = BasicMatrixMap<const T>;
112 struct Index3 {
121 struct Index {
126 SparseSlice(int nrows, int ncols, int bsize) in SparseSlice() argument
141 std::vector<int> index3_offset;
142 std::vector<Index3> index3;
143 std::vector<T> data3;
147 std::vector<int> index_offset;
148 std::vector<Index> index;
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go46275 func SparseSlice(scope *Scope, indices tf.Output, values tf.Output, shape tf.Output, start tf.Outpu… func