Searched defs:sparse_indices (Results 1 – 10 of 10) sorted by relevance
31 def _SparseToDense(sparse_indices, argument
71 auto sparse_indices = sparse_indices_out->tensor<int64, 2>(); in Compute() local
216 Tensor sparse_indices(DT_INT64, TensorShape({N, NDIM})); in BM_SparseToDense() local
160 std::vector<NodeBuilder::NodeOut> sparse_indices; in GetGraphs() local
258 OpOutputList sparse_indices; in WriteOutput() local370 OpOutputList sparse_indices; in Compute() local
119 std::vector<Tensor> sparse_indices; member
845 def pad_sparse_embedding_lookup_indices(sparse_indices, padded_size): argument
414 repeated int64 sparse_indices = 14; field
1181 def sparse_to_dense(sparse_indices, argument
66 std::vector<TfLiteTensor*> sparse_indices; member