Searched refs:forward_from (Results 1 – 4 of 4) sorted by relevance
271 int forward_from = 0; in DoReduce() local272 op_params.forward_from_array = &forward_from; in DoReduce()
207 const int* forward_from() const { return forward_from_base(); } in forward_from() function518 int* forward_from = item->forward_from_base(); in InitializeNode() local524 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()
652 int forward_from[] = {OpKernelContext::Params::kNeverForward}; in DoBroadcast() local653 if (forward_input) forward_from[0] = 0; in DoBroadcast()655 op_params.forward_from_array = &forward_from[0]; in DoBroadcast()
493 int forward_from = 0; in DoReduce() local494 op_params.forward_from_array = &forward_from; in DoReduce()