Searched refs:forward_from_array (Results 1 – 10 of 10) sorted by relevance
511 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()512 params_->forward_from_array[output_index] == Params::kNeverForward); in forward_input()515 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()516 params_->forward_from_array[output_index] == input_index); in forward_input()517 if (!forward_expected && params_->forward_from_array != nullptr) { in forward_input()520 if (params_->forward_from_array[i] == input_index) { in forward_input()666 (params_->forward_from_array != nullptr && index >= 0 && in allocate_output()667 params_->forward_from_array[index] >= 0); in allocate_output()905 (params_->forward_from_array != nullptr && in maybe_set_output_by_allocate_and_copy()906 params_->forward_from_array[index] == Params::kNeverForward); in maybe_set_output_by_allocate_and_copy()[all …]
699 const int* forward_from_array = nullptr; member
458 params.forward_from_array = forward_from.data(); in TEST_F()
93 sub_params_.forward_from_array = &forward_from_;
503 op_params.forward_from_array = &forward_from; in DoReduce()
653 op_params.forward_from_array = &forward_from[0]; in DoBroadcast()
804 params.forward_from_array = item.forward_from(); in Process()
315 params.forward_from_array = nullptr; in Run()
311 op_params.forward_from_array = &forward_from; in RunReduce()
275 params->forward_from_array = {}; in SetOutputAttrs()