Searched refs:output_indices_t (Results 1 – 6 of 6) sorted by relevance
97 Tensor* output_indices_t; in Compute() local101 &output_indices_t)); in Compute()139 Tensor* output_indices_t; in Compute() local144 &output_indices_t)); in Compute()145 auto output_indices = output_indices_t->matrix<int64>(); in Compute()
194 Tensor *output_indices_t, *output_values_t; in Compute() local197 &output_indices_t)); in Compute()200 auto output_indices_mat = output_indices_t->matrix<int64>(); in Compute()
298 auto output_indices_t = output_indices.matrix<int64>(); in Compute() local303 output_indices_t(i, d - 1) = indices(i, d); in Compute()429 const auto& output_indices_t = output_indices.matrix<int64>(); in Compute() local435 expanded_indices_t.slice(indices_start, indices_sizes) = output_indices_t; in Compute()
117 const auto& output_indices_t = output_indices.matrix<int64>(); in Compute() local124 output_indices_t; in Compute()
243 auto output_indices_t = output_indices.matrix<int64>(); in Compute() local248 output_indices_t(i, d - 1) = indices(i, d); in Compute()
582 std::vector<typename TTypes<int64>::Matrix> output_indices_t; in Split() local584 output_indices_t.reserve(num_split); in Split()619 output_indices_t.emplace_back(output_indices[i].matrix<int64>()); in Split()633 output_indices_t[slice_index](slice_dim, j) = in Split()699 auto output_indices_t = output_indices.matrix<int64>(); in Slice() local721 output_indices_t(index, dim) = input_indices_t(i, dim) - start[dim]; in Slice()