Lines Matching defs:output_index
403 int output_index) {
407 set_output_ref(output_index, (*params_->inputs)[input_index].mutex_if_ref,
412 int input_index, int output_index, const TensorShape& output_shape,
416 : output_alloc_attr(output_index);
418 input_index, expected_output_dtype(output_index), output_shape,
419 output_memory_type(output_index), output_attr);
422 outputs_[output_index] = TensorValue(new_tensor.release());
423 *output = outputs_[output_index].tensor;
433 int input_index, output_index, stop;
443 params_->op_kernel->OutputRange(output_name, &output_index
444 if (stop != output_index + 1) {
450 if (!forward_input_to_output_with_shape(input_index, output_index,
