Home
last modified time | relevance | path

Searched refs:num_return_values (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc329 size_t num_return_values; in RestoreInternal() local
333 num_return_values = static_cast<size_t>(size); in RestoreInternal()
334 if (num_return_values != size) { in RestoreInternal()
340 result.return_values.reserve(num_return_values); in RestoreInternal()
341 for (size_t j = 0; j < num_return_values; j++) { in RestoreInternal()
Dparallel_interleave_dataset_op.cc1268 int64 num_return_values; in ReadElement() local
1272 &num_return_values)); in ReadElement()
1273 result->return_values.reserve(num_return_values); in ReadElement()
1274 for (size_t j = 0; j < num_return_values; j++) { in ReadElement()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc479 size_t num_return_values; in RestoreInternal() local
486 num_return_values = static_cast<size_t>(size); in RestoreInternal()
487 if (num_return_values != size) { in RestoreInternal()
494 result.return_values.reserve(num_return_values); in RestoreInternal()
495 for (size_t j = 0; j < num_return_values; j++) { in RestoreInternal()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1304 unsigned num_return_values = inst->getNumOperands(); in Lower() local
1305 if ((return_tuple_ && is_entry_function) || num_return_values != 1) { in Lower()
1309 std::vector<xla::XlaOp> returns(num_return_values); in Lower()
1341 } else if (num_return_values == 1) { in Lower()