Searched defs:sp_shape (Results 1 – 4 of 4) sorted by relevance
224 auto sp_shape = sp_shape_t->vec<int64_t>(); in Compute() local294 auto sp_shape = sp_shape_t->vec<int64_t>(); in Compute() local
87 auto sp_shape = sp.shape(); in SparseTensorReduceHelper() local
214 Node* sp_shape, Node* dense) { in SparseMatCMulDenseMat()
1486 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None): argument