Searched refs:output_sampled_index_tensor (Results 1 – 1 of 1) sorted by relevance
267 Tensor* output_sampled_index_tensor; in Compute() local270 &output_sampled_index_tensor)); in Compute()271 auto output = output_sampled_index_tensor->scalar<int64_t>(); in Compute()