Searched refs:sp_indices_d (Results 1 – 2 of 2) sorted by relevance
1039 Tensor* sp_indices_d = nullptr; in Compute() local1042 &sp_indices_d)); in Compute()1048 auto indices_t = sp_indices_d->matrix<int64>(); in Compute()1053 Tensor* sp_indices_d = nullptr; in Compute() local1057 &sp_indices_d)); in Compute()1163 Tensor* sp_indices_d = nullptr; in Compute() local1167 &sp_indices_d)); in Compute()1180 sparse_values_tmp[t], t, offset, sp_indices_d, sp_values_d); in Compute()
395 Tensor* sp_indices_d = &(*output_sparse_indices_tensor)[d]; in BatchExampleProtoToTensors() local401 sparse_values_tensor[b], b, offset, sp_indices_d, sp_values_d); in BatchExampleProtoToTensors()