Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_quantize_op.cc64 memory::desc dst_md; member
410 memory::desc dst_md = in Compute() local
H A Dmkl_dequantize_op.cc113 memory::desc dst_md = in Compute() local
H A Dmkl_concat_op.cc332 std::shared_ptr<dnnl::memory::desc> dst_md; member
676 memory::desc dst_md({}, memory::data_type::undef, in Compute() local
H A Dmkl_pooling_ops_common.h129 std::shared_ptr<dnnl::memory::desc> dst_md; member
270 std::shared_ptr<dnnl::memory::desc> dst_md; member
H A Dmkl_matmul_op_fused.cc182 auto dst_md = in Compute() local
H A Dmkl_eltwise_activation_base_op.h127 std::shared_ptr<memory::desc> dst_md; member
H A Dmkl_conv_ops.cc238 std::shared_ptr<dnnl::memory::desc> dst_md; member
1091 auto dst_md = conv_prim_desc.dst_desc(); in AllocateOutputTensor() local
2027 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor() local
H A Dmkl_matmul_ops_common.h185 std::shared_ptr<dnnl::memory::desc> dst_md; member
H A Dmkl_relu_op.cc118 std::shared_ptr<memory::desc> dst_md; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/
H A DMatmul.cpp117 dnnl::memory::desc dst_md, dst_usr_md, dst_any_md; in matmul() local