Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_dequantize_op.cc95 memory::desc dst_md = memory::desc(); in Compute() local
Dmkl_quantize_op.cc62 memory::desc dst_md; member
401 memory::desc dst_md = in Compute() local
Dmkl_matmul_op_fused.cc168 auto dst_md = in Compute() local
Dmkl_concat_op.cc325 std::shared_ptr<mkldnn::memory::desc> dst_md; member
664 memory::desc dst_md({}, memory::data_type::undef, in Compute() local
Dmkl_pooling_ops_common.h127 std::shared_ptr<mkldnn::memory::desc> dst_md; member
264 std::shared_ptr<mkldnn::memory::desc> dst_md; member
Dmkl_conv_ops.cc187 std::shared_ptr<mkldnn::memory::desc> dst_md; member
855 auto dst_md = conv_prim_desc.dst_desc(); in AllocateOutputTensor() local
1754 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor() local
Dmkl_matmul_ops_common.h154 std::shared_ptr<mkldnn::memory::desc> dst_md; member
Dmkl_relu_op.cc112 std::shared_ptr<memory::desc> dst_md; member
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h111 #define REORDER_PD_CONSTRUCTOR(src_md, dst_md, engine) \ argument
113 #define REORDER_PD_CONSTRUCTOR_WITH_ATTR(src_md, dst_md, engine, prim_attr) \ argument