Searched refs:output_nums (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 174 const std::vector<int32>& output_nums, TensorHandle** retvals, in AddOpRetvalsToResponse() argument
195 const int output_num = output_nums.empty() ? i : output_nums.at(i); in AddOpRetvalsToResponse()
474 std::vector<int32> output_nums; in RunComponentFunction() local
476 output_nums.push_back(output_num); in RunComponentFunction()
486 [op, op_id = operation.id(), num_retvals, retvals, output_nums, cm, in RunComponentFunction()
504 eager_context, op_id, *num_retvals, output_nums, retvals->data(), in RunComponentFunction()
|