Home
last modified time | relevance | path

Searched refs:forward_from (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc271 int forward_from = 0; in DoReduce() local
272 op_params.forward_from_array = &forward_from; in DoReduce()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc207 const int* forward_from() const { return forward_from_base(); } in forward_from() function
518 int* forward_from = item->forward_from_base(); in InitializeNode() local
524 forward_from[i] = OpKernelContext::Params::kNoReservation; in InitializeNode()
530 forward_from[i] = OpKernelContext::Params::kNeverForward; in InitializeNode()
537 forward_from[i] == OpKernelContext::Params::kNoReservation) { in InitializeNode()
541 DCHECK_EQ(forward_from[i], OpKernelContext::Params::kNoReservation); in InitializeNode()
542 forward_from[i] = forward_input[j]; in InitializeNode()
1735 params.forward_from_array = item.forward_from(); in Process()
Dhierarchical_tree_broadcaster_test.cc652 int forward_from[] = {OpKernelContext::Params::kNeverForward}; in DoBroadcast() local
653 if (forward_input) forward_from[0] = 0; in DoBroadcast()
655 op_params.forward_from_array = &forward_from[0]; in DoBroadcast()
Dring_reducer_test.cc493 int forward_from = 0; in DoReduce() local
494 op_params.forward_from_array = &forward_from; in DoReduce()