Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/
H A DMatmul.cpp118 dnnl::memory::desc bias_md; in matmul() local
H A DDeconv.cpp261 auto bias_md = bias_defined in deconvolution_backward_data() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.cc237 std::shared_ptr<dnnl::memory::desc> bias_md; member
1818 auto bias_md = memory::desc({static_cast<int>(bias_tensor.NumElements())}, in GetBiasHandle() local
H A Dmkl_matmul_ops_common.h184 std::shared_ptr<dnnl::memory::desc> bias_md; member