Searched refs:input_split (Results 1 – 2 of 2) sorted by relevance
53 …NodeID input_split = GraphBuilder::add_split_node(g, params, input, num_grou… in create_grouped_convolution() local73 g.add_connection(input_split, i, conv_nid, 0); in create_grouped_convolution()
263 std::vector<absl::string_view> input_split = in IsEinsumCorrectlyFormed() local274 absl::string_view a_input_str = input_split[0]; in IsEinsumCorrectlyFormed()275 absl::string_view b_input_str = input_split[1]; in IsEinsumCorrectlyFormed()1153 std::vector<absl::string_view> input_split = in GenerateBatchMatmulContextFromEinsum() local1158 absl::string_view a_input_str = input_split[0]; in GenerateBatchMatmulContextFromEinsum()1159 absl::string_view b_input_str = input_split[1]; in GenerateBatchMatmulContextFromEinsum()