Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/mkldnn/
DConvPrepack.cpp149 const ideep::tensor& mkldnn_weight, in mkldnn_convolution_out()
184 const ideep::tensor& mkldnn_weight = context.weight_packed_; in get_output_sizes() local