Searched refs:forward_from_array (Results 1 – 9 of 9) sorted by relevance
524 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()525 params_->forward_from_array[output_index] == Params::kNeverForward); in forward_input()528 (params_->forward_from_array != nullptr && output_index >= 0 && in forward_input()529 params_->forward_from_array[output_index] == input_index); in forward_input()530 if (!forward_expected && params_->forward_from_array != nullptr) { in forward_input()533 if (params_->forward_from_array[i] == input_index) { in forward_input()710 (params_->forward_from_array != nullptr && index >= 0 && in allocate_output()711 params_->forward_from_array[index] >= 0); in allocate_output()916 (params_->forward_from_array != nullptr && in maybe_set_output_by_allocate_and_copy()917 params_->forward_from_array[index] == Params::kNeverForward); in maybe_set_output_by_allocate_and_copy()[all …]
671 const int* forward_from_array = nullptr; member926 return params_->forward_from_array != nullptr && in output_expects_forwarding()927 params_->forward_from_array[index] >= 0; in output_expects_forwarding()
461 params.forward_from_array = forward_from.data(); in TEST_F()
94 sub_params_.forward_from_array = &forward_from_;
338 op_params.forward_from_array = &forward_from; in RunCollective()
344 params.forward_from_array = nullptr; in Run()
857 params.forward_from_array = item.forward_from(); in ProcessInline()
313 op_params.forward_from_array = &forward_from; in RunReduce()
275 params->forward_from_array = {}; in SetOutputAttrs()