Searched refs:expected_max_outputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 90 const TFE_OpAttrs* attributes, int expected_max_outputs,
101 const TFE_OpAttrs* attributes, int expected_max_outputs,
179 int expected_max_outputs, in StartExecute() argument
192 expected_max_outputs_ = expected_max_outputs; in StartExecute()
224 int expected_max_outputs, in Execute() argument
242 std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs); in Execute()
243 int real_num_outputs = expected_max_outputs; in Execute()
300 const TFE_OpAttrs* attributes, int expected_max_outputs, in Execute() argument
302 std::vector<PartialTensorShape> expected_output_shapes(expected_max_outputs); in Execute()
304 expected_max_outputs, *default_cancellation_manager_); in Execute()
[all …]
|
| D | parallel_device_lib.h | 98 int expected_max_outputs, TF_Status* status) const;
115 int expected_max_outputs,
|
| D | parallel_device.cc | 68 const TFE_OpAttrs* attributes, int expected_max_outputs, in ExecuteWithSpecialOps() argument
167 attributes, expected_max_outputs, status)); in ExecuteWithSpecialOps()
|