Searched refs:forward_from (Results 1 – 7 of 7) sorted by relevance
210 int* forward_from = item->forward_from_base(); in InitializeNode() local216 forward_from[i] = OpKernelContext::Params::kNoReservation; in InitializeNode()222 forward_from[i] = OpKernelContext::Params::kNeverForward; in InitializeNode()229 forward_from[i] == OpKernelContext::Params::kNoReservation) { in InitializeNode()233 DCHECK_EQ(forward_from[i], OpKernelContext::Params::kNoReservation); in InitializeNode()234 forward_from[i] = forward_input[j]; in InitializeNode()
143 const int* forward_from() const { return forward_from_base(); } in forward_from() function
650 int forward_from[] = {OpKernelContext::Params::kNeverForward}; in DoBroadcast() local651 if (forward_input) forward_from[0] = 0; in DoBroadcast()653 op_params.forward_from_array = &forward_from[0]; in DoBroadcast()
502 int forward_from = 0; in DoReduce() local503 op_params.forward_from_array = &forward_from; in DoReduce()
804 params.forward_from_array = item.forward_from(); in Process()
310 int forward_from = 0; in RunReduce() local311 op_params.forward_from_array = &forward_from; in RunReduce()
457 std::vector<int> forward_from({OpKernelContext::Params::kNeverForward}); in TEST_F() local458 params.forward_from_array = forward_from.data(); in TEST_F()