Searched refs:expanded_indices (Results 1 – 2 of 2) sorted by relevance
116 Tensor expanded_indices(DT_INT64, TensorShape({num_entries, 1 + rank})); in Compute() local118 auto expanded_indices_t = expanded_indices.matrix<int64>(); in Compute()134 indices.push_back(expanded_indices); in Compute()
426 Tensor expanded_indices( in Compute() local430 auto expanded_indices_t = expanded_indices.matrix<int64>(); in Compute()444 indices_to_concat.push_back(std::move(expanded_indices)); in Compute()