Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc541 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()
542 params_->forward_from_array[output_index] == Params::kNeverForward); in forward_input()
545 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()
546 params_->forward_from_array[output_index] == input_index); in forward_input()
547 if (!forward_expected && params_->forward_from_array != nullptr) { in forward_input()
550 if (params_->forward_from_array[i] == input_index) { in forward_input()
707 (params_->forward_from_array != nullptr && index >= 0 && in allocate_output()
708 params_->forward_from_array[index] >= 0); in allocate_output()
918 (params_->forward_from_array != nullptr && in set_output()
919 params_->forward_from_array[index] == Params::kNeverForward); in set_output()
[all …]
Dop_kernel.h730 const int* forward_from_array = nullptr; member
Dop_kernel_test.cc515 params.forward_from_array = forward_from.data(); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc93 sub_params_.forward_from_array = &forward_from_;
Dring_reducer_test.cc499 op_params.forward_from_array = &forward_from; in DoReduce()
Dhierarchical_tree_broadcaster_test.cc660 op_params.forward_from_array = &forward_from[0]; in DoBroadcast()
Dexecutor.cc1826 params.forward_from_array = item.forward_from(); in Process()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc270 params.forward_from_array = nullptr; in Run()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc309 op_params.forward_from_array = &forward_from; in RunReduce()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr_test.cc420 params->forward_from_array = {}; in SetOutputAttrs()