Searched refs:input_split (Results 1 – 3 of 3) 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()
264 std::vector<absl::string_view> input_split = in IsEinsumCorrectlyFormed() local275 absl::string_view a_input_str = input_split[0]; in IsEinsumCorrectlyFormed()276 absl::string_view b_input_str = input_split[1]; in IsEinsumCorrectlyFormed()1154 std::vector<absl::string_view> input_split = in GenerateBatchMatmulContextFromEinsum() local1159 absl::string_view a_input_str = input_split[0]; in GenerateBatchMatmulContextFromEinsum()1160 absl::string_view b_input_str = input_split[1]; in GenerateBatchMatmulContextFromEinsum()
6263 dnn::BatchDescriptor input_split = input_dimensions; in DoPoolForward() local6269 input_split.set_count(splits[i].num_batches); in DoPoolForward()6271 CudnnTensorDescriptor src_desc(input_split, cudnn_input_type); in DoPoolForward()6332 dnn::BatchDescriptor input_split = input_dimensions; in DoPoolBackward() local6338 input_split.set_count(splits[i].num_batches); in DoPoolBackward()6340 CudnnTensorDescriptor src_desc(input_split, cudnn_input_type); in DoPoolBackward()