Searched refs:sp_indices_d (Results 1 – 2 of 2) sorted by relevance
683 Tensor* sp_indices_d = nullptr; in Compute() local686 &sp_indices_d)); in Compute()692 auto indices_t = sp_indices_d->matrix<int64>(); in Compute()697 Tensor* sp_indices_d = nullptr; in Compute() local701 &sp_indices_d)); in Compute()807 Tensor* sp_indices_d = nullptr; in Compute() local811 &sp_indices_d)); in Compute()824 sparse_values_tmp[t], t, offset, sp_indices_d, sp_values_d); in Compute()
392 Tensor* sp_indices_d = &(*output_sparse_indices_tensor)[d]; in BatchExampleProtoToTensors() local398 sparse_values_tensor[b], b, offset, sp_indices_d, sp_values_d); in BatchExampleProtoToTensors()