Searched refs:input_split (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | GroupedConvolutionMutator.cpp | 53 …NodeID input_split = GraphBuilder::add_split_node(g, params, input, num_grou… in create_grouped_convolution() local
73 g.add_connection(input_split, i, conv_nid, 0); in create_grouped_convolution()
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_fake_pg.py | 147 input_split = [1, 1]
148 dist.all_to_all_single(out_tensor, in_tensor, output_split, input_split)
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 264 std::vector<absl::string_view> input_split = in IsEinsumCorrectlyFormed() local
275 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() local
1159 absl::string_view a_input_str = input_split[0]; in GenerateBatchMatmulContextFromEinsum()
1160 absl::string_view b_input_str = input_split[1]; in GenerateBatchMatmulContextFromEinsum()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 6263 dnn::BatchDescriptor input_split = input_dimensions; in DoPoolForward() local
6269 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() local
6338 input_split.set_count(splits[i].num_batches); in DoPoolBackward()
6340 CudnnTensorDescriptor src_desc(input_split, cudnn_input_type); in DoPoolBackward()
|