Searched refs:output_index_t (Results 1 – 1 of 1) sorted by relevance
43 const Tensor& output_index_t = context->input(1); in Compute() local44 OP_REQUIRES(context, TensorShapeUtils::IsScalar(output_index_t.shape()), in Compute()47 output_index_t.shape().DebugString())); in Compute()48 int output_index = output_index_t.scalar<int>()(); in Compute()