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