Home
last modified time | relevance | path

Searched defs:kSpatialDim (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dunpack_quantized_weights.cpp324 const int64_t kSpatialDim = config_vals.at(0); in unpackQuantizedWeightsHelper() local
383 const int64_t kSpatialDim = conv_params_packed[0].item<int64_t>(); in unpackQuantizedWeightsHelper() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqconv.cpp1030 int kSpatialDim, in can_use_xnnp()
1430 int kSpatialDim = act.dim() - 2; in _quantized_convolution_onednn() local
H A Dqconv_prepack.cpp503 int kSpatialDim = weight.ndimension() - 2; in _qconv_prepack_onednn() local