Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc111 struct SparseSlice { struct
112 using ConstMatrixMap = BasicMatrixMap<const T>;
116 struct Index3 {
125 struct Index {
130 SparseSlice(int nrows, int ncols, int bsize) in SparseSlice() function
145 std::vector<int> index3_offset;
146 std::vector<Index3> index3;
147 std::vector<T> data3;
151 std::vector<int> index_offset;
152 std::vector<Index> index;
[all …]
/external/tensorflow/tensorflow/go/op/
Dwrappers.go32230 func SparseSlice(scope *Scope, indices tf.Output, values tf.Output, shape tf.Output, start tf.Outpu… func