Searched refs:src_output_index (Results 1 – 3 of 3) sorted by relevance
82 map<int64, int32> src_output_index = 14; field
478 for (const auto& s : node.src_output_index()) { in FromProto()
866 for (int src_output_index = 0; in CreateOrGetSplitNodesForInputSharding() local867 src_output_index < input_split_node->num_outputs(); in CreateOrGetSplitNodesForInputSharding()868 ++src_output_index) { in CreateOrGetSplitNodesForInputSharding()874 src_output_index, dtype, in CreateOrGetSplitNodesForInputSharding()