Home
last modified time | relevance | path

Searched defs:input_contig (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_dynamic.cpp49 auto input_contig = input.contiguous(); in apply_dynamic_impl() local
258 auto input_contig = input.contiguous(); in apply_dynamic_impl() local
413 const at::Tensor input_contig = input.contiguous(); in apply_dynamic_impl() local
508 auto input_contig = input.contiguous(); in apply_dynamic_impl() local
H A Dqlinear.cpp56 auto input_contig = input.expect_contiguous(); in apply_impl() local
318 auto input_contig = input.expect_contiguous(); in apply_with_input_q_dq_qweight_dq_output_fp32_impl() local
471 const auto input_contig = input.contiguous(); in apply_impl_xnnp() local
623 auto input_contig = input.contiguous(); in apply_impl() local
803 auto input_contig = input.expect_contiguous(); in apply_impl() local
982 auto input_contig = in linear_int8_with_onednn_weight() local
H A DAveragePool2d.cpp51 Tensor input_contig = input.contiguous(); in avg_pool2d_out_frame() local
292 Tensor input_contig = input.contiguous(c10::MemoryFormat::ChannelsLast); in qnnpack_avg_pool2d() local
H A Dqtanh.cpp38 Tensor input_contig = input.contiguous(input.suggest_memory_format()); in qnnpack_tanh() local
H A Dqhardsigmoid.cpp36 Tensor input_contig = input.contiguous(input.suggest_memory_format()); in qnnpack_hardsigmoid() local
H A Dqsigmoid.cpp41 Tensor input_contig = input.contiguous(input.suggest_memory_format()); in qnnpack_sigmoid() local
H A Dqclamp.cpp40 Tensor input_contig = input.contiguous(input.suggest_memory_format()); in qnnpack_clamp() local
H A DReduceOps.cpp76 Tensor input_contig = input.contiguous(MemoryFormat::ChannelsLast); in qnnpack_mean() local
H A Dqrelu.cpp49 Tensor input_contig = input.contiguous(input.suggest_memory_format()); in qnnpack_relu() local
H A DUpSampleBilinear2d.cpp199 auto input_contig = input.contiguous(); in upsample_bilinear2d_quantized_cpu() local
H A DUpSampleNearest2d.cpp181 auto input_contig = input.contiguous(); in _upsample_nearest2d_quantized_cpu() local
H A DUpSampleNearest3d.cpp193 auto input_contig = input.contiguous(); in _upsample_nearest3d_quantized_cpu() local
H A DAdaptiveAveragePooling.cpp228 auto input_contig = input.contiguous(); in _adaptive_avg_pool() local
H A DPooling.cpp530 Tensor input_contig = input.contiguous(MemoryFormat::ChannelsLast); in qnnpack_maxpool2d() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DGridSampler.cpp88 auto input_contig = contiguousIfZeroInStrides(input_t); in cudnn_grid_sampler_forward() local
138 auto input_contig = contiguousIfZeroInStrides(input_t); in cudnn_grid_sampler_backward() local
H A DConvShared.cpp225 Tensor input_contig = input->contiguous(memory_format); in cudnn_convolution_forward_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DQTensor.cpp21 auto input_contig = self.contiguous(); in quantize_per_tensor_dynamic() local
274 auto input_contig = self.contiguous(); in _choose_qparams_per_tensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqlinear.cpp39 auto input_contig = input.contiguous(); in apply_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DConv_miopen.cpp785 Tensor input_contig = input->contiguous(memory_format); in miopen_convolution_forward() local
888 Tensor input_contig = input->contiguous(memory_format); in miopen_depthwise_convolution_forward() local
1083 TensorArg input_contig{ input_contig_t, "input", 2}; in miopen_depthwise_convolution_backward_weight() local
1136 TensorArg input_contig{ input_contig_t, "input", 2}; in miopen_convolution_backward_weight() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DQuantizedLinear.cpp426 const Tensor input_contig = input.contiguous(); in fbgemm_linear_fp16_weight_fp32_activation() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp2714 const auto input_contig = input.expect_contiguous(); in signed_log1p_out() local