Home
last modified time | relevance | path

Searched refs:forward_from_array (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc509 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()
510 params_->forward_from_array[output_index] == Params::kNeverForward); in forward_input()
513 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()
514 params_->forward_from_array[output_index] == input_index); in forward_input()
515 if (!forward_expected && params_->forward_from_array != nullptr) { in forward_input()
518 if (params_->forward_from_array[i] == input_index) { in forward_input()
662 (params_->forward_from_array != nullptr && index >= 0 && in allocate_output()
663 params_->forward_from_array[index] >= 0); in allocate_output()
Dop_kernel.h679 const int* forward_from_array = nullptr; member
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc95 sub_params_.forward_from_array = &forward_from_;
Dring_reducer_test.cc494 op_params.forward_from_array = &forward_from; in DoReduce()
Dhierarchical_tree_broadcaster_test.cc655 op_params.forward_from_array = &forward_from[0]; in DoBroadcast()
Dexecutor.cc1735 params.forward_from_array = item.forward_from(); in Process()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc236 params.forward_from_array = nullptr; in RunAsync()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc272 op_params.forward_from_array = &forward_from; in DoReduce()