Searched defs:output_cpu (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | rnn.cpp | 586 auto output_cpu = gru_cpu->forward(input_cpu); in TEST_F() local
640 auto output_cpu = lstm_cpu->forward(input_cpu); in TEST_F() local
697 auto output_cpu = lstm_cpu->forward(input_cpu); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 925 …const auto output_cpu = at::batch_norm(input_cpu, weight_cpu, bias_cpu, running_mean_cpu, running_… in TEST_F() local
954 …const auto output_cpu = at::batch_norm(input_cpu, weight_cpu, bias_cpu, running_mean_cpu, running_… in TEST_F() local
984 …const auto output_cpu = at::batch_norm(input_cpu, weight_cpu, bias_cpu, running_mean_cpu, running_… in TEST_F() local
1401 const auto output_cpu = at::conv1d( in TEST_F() local
1439 const auto output_cpu = at::conv1d( in test_conv1d() local
1641 const auto output_cpu = at::conv2d( in TEST_F() local
1734 const auto output_cpu = at::conv2d( in TEST_F() local
1799 const auto output_cpu = at::conv2d( in TEST_F() local
1861 const auto output_cpu = at::conv2d( in TEST_F() local
1951 const auto output_cpu = at::conv2d( in TEST_F() local
[all …]
|
| H A D | vulkan_quantized_api_test.cpp | 1519 const auto output_cpu = at::conv2d( in test_conv2d() local
1627 const auto output_cpu = at::conv2d( in TEST_F() local
1730 const auto output_cpu = at::conv2d( in TEST_F() local
2286 at::Tensor output_cpu = in test_quantized_binary_op() local
2617 const auto output_cpu = at::conv2d( in test_quantized_conv2d() local
|
| H A D | xnnpack_test.cpp | 144 const auto output_cpu = at::conv2d( in TEST() local
|