Searched refs:inputs_to_sharded_retval (Results 1 – 1 of 1) sorted by relevance
1027 std::vector<NodeOut> inputs_to_sharded_retval = orig_inputs; in CreateConcatNodesForRetval() local1035 int num_concat_nodes = inputs_to_sharded_retval.size() / num_splits; in CreateConcatNodesForRetval()1041 inputs_to_sharded_retval.begin() + input_index_to_concat_node; in CreateConcatNodesForRetval()1056 inputs_to_sharded_retval = new_concat_nodes; in CreateConcatNodesForRetval()1059 TF_RET_CHECK(inputs_to_sharded_retval.size() == 1); in CreateConcatNodesForRetval()1063 inputs_to_sharded_retval.at(0).node, in CreateConcatNodesForRetval()1067 return inputs_to_sharded_retval.at(0).node; in CreateConcatNodesForRetval()