Home
last modified time | relevance | path

Searched defs:input_n (Results 1 – 11 of 11) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DIm2Col.cpp99 Tensor input_n; in im2col_out_cpu_template() local
DCol2Im.cpp149 Tensor input_n = Tensor(); in col2im_out_cpu_template() local
DNaiveConvolutionTranspose2d.cpp311 Tensor input_n = input_.select(0, elt); in slow_conv_transpose2d_out_cpu_template() local
694 Tensor input_n = Tensor(); in slow_conv_transpose2d_acc_grad_parameters_cpu() local
DNaiveConvolutionTranspose3d.cpp296 Tensor input_n; in slow_conv_transpose3d_out_cpu_template() local
735 Tensor input_n; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
DNaiveDilatedConvolution.cpp227 Tensor input_n = input.select(0, elt); in slow_conv_dilated_all_cpu_template() local
/external/pytorch/aten/src/ATen/native/cuda/
DIm2Col.cu108 Tensor input_n; in im2col_out_cuda_template() local
DConvolutionMM2d.cu177 auto input_n = input.select(0, elt); in slow_conv2d_forward() local
334 auto input_n = input.select(0, elt); in slow_conv2d_grad_weight() local
DNaiveConvolutionTranspose2d.cu217 Tensor input_n; in slow_conv_transpose2d_out_cuda_template() local
607 Tensor input_n = Tensor(); in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
DNaiveConvolutionTranspose3d.cu299 Tensor input_n; in slow_conv_transpose3d_out_cuda_template() local
752 Tensor input_n; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
DNaiveDilatedConvolution.cu262 Tensor input_n = input.select(0, elt); in slow_conv_dilated_all_cuda_template() local
/external/vixl/test/aarch64/
Dtest-simulator-aarch64.cc3078 #define CALL_TEST_NEON_HELPER_1Op(mnemonic, vdform, vnform, input_n) \ argument
3088 #define CALL_TEST_NEON_HELPER_1OpAcross(mnemonic, vdform, vnform, input_n) \ argument
3104 input_n, \ argument
3122 input_n, \ argument
3140 input_n, \ argument
3165 input_n, \ argument
3193 input_n, \ argument
3314 #define CALL_TEST_NEON_HELPER_ACROSS(mnemonic, vd_form, vn_form, input_n) \ argument
3362 #define CALL_TEST_NEON_HELPER_2DIFF(mnemonic, vdform, vnform, input_n) \ argument
3627 #define DEFINE_TEST_NEON_FHM(mnemonic, input_d, input_n, input_m) \ argument
[all …]