Searched refs:output_indices_data (Results 1 – 2 of 2) sorted by relevance
169 int64* output_indices_data = output_indices->matrix<int64>().data(); in Compute() local214 *output_indices_data++ = i; in Compute()215 *output_indices_data++ = *element_indices_data++; in Compute()237 *output_indices_data++ = sparse_tensor_index_component; in Compute()240 *output_indices_data++ = *element_indices_data++; in Compute()
222 TTypes<int, 1>::Tensor output_indices_data = output_indices->tensor<int, 1>(); in DoNonMaxSuppressionOp() local223 std::copy_n(selected.begin(), selected.size(), output_indices_data.data()); in DoNonMaxSuppressionOp()