Searched defs:sparse_indices (Results 1 – 14 of 14) sorted by relevance
27 def _SparseToDense(sparse_indices, argument
74 auto sparse_indices = sparse_indices_out->tensor<int64_t, 2>(); in Compute() local
216 Tensor sparse_indices(DT_INT64, TensorShape({N, NDIM})); in BM_SparseToDense() local
258 OpOutputList sparse_indices; in WriteOutput() local370 OpOutputList sparse_indices; in Compute() local
161 std::vector<NodeBuilder::NodeOut> sparse_indices; in GetGraphs() local
638 def create_num_blocks_fake_generator(sparse_indices): argument
119 std::vector<Tensor> sparse_indices; member
463 const std::optional<at::Tensor>& sparse_indices, in allreduce_sparse_cuda_()
262 auto sparse_indices = offsets.convert<ResTable_sparseTypeEntry>().iterator(); in GetEntryOffset() local
845 def pad_sparse_embedding_lookup_indices(sparse_indices, padded_size): argument
462 repeated int64 sparse_indices = 14; field
1174 def sparse_to_dense(sparse_indices, argument
68 std::vector<TfLiteTensor*> sparse_indices; member
2024 Tensor sparse_indices = indices_dim1_dim2.slice(1, mat_el_begin_idx, mat_el_end_idx); in bmm_out_sparse_cpu() local