Searched refs:forward_from_array (Results 1 – 10 of 10) sorted by relevance
541 (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 …]
730 const int* forward_from_array = nullptr; member
515 params.forward_from_array = forward_from.data(); in TEST_F()
93 sub_params_.forward_from_array = &forward_from_;
499 op_params.forward_from_array = &forward_from; in DoReduce()
660 op_params.forward_from_array = &forward_from[0]; in DoBroadcast()
1826 params.forward_from_array = item.forward_from(); in Process()
270 params.forward_from_array = nullptr; in Run()
309 op_params.forward_from_array = &forward_from; in RunReduce()
420 params->forward_from_array = {}; in SetOutputAttrs()