| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_dequantize_op.cc | 95 memory::desc dst_md = memory::desc(); in Compute() local
97 dst_md = memory::desc(src_mkl_shape.GetMklLayout().data); in Compute()
100 dst_md.data.data_type = memory::convert_to_c(MklDnnType<float>()); in Compute()
102 dst_md = memory::desc(src_dims, MklDnnType<float>(), in Compute()
110 output_mkl_shape.SetMklLayout(&dst_md); in Compute()
115 output_tf_shape.AddDim(dst_md.get_size() / sizeof(float)); in Compute()
124 dst.SetUsrMem(dst_md, output_tensor); in Compute()
|
| D | mkl_quantize_op.cc | 62 memory::desc dst_md; member
71 memory::desc dst_md) in MklReorderWithScaleFwdParams()
72 : src_dims(src_dims), src_md(src_md), dst_md(dst_md) {} in MklReorderWithScaleFwdParams()
132 new memory(fwdParams.dst_md, cpu_engine_, DummyData)); in Setup()
401 memory::desc dst_md = in Compute() local
409 output_mkl_shape.SetMklLayout(&dst_md); in Compute()
414 output_tf_shape.AddDim(dst_md.get_size() / sizeof(T)); in Compute()
423 dst.SetUsrMem(dst_md, output_tensor); in Compute()
468 MklReorderWithScaleFwdParams fwdParams(src_dims, src_md, dst_md); in Compute()
|
| D | mkl_concat_op.cc | 325 std::shared_ptr<mkldnn::memory::desc> dst_md; member
353 context_.dst_md.reset(new memory::desc({concat_fwd_dims.dst_dims}, in Setup()
358 *context_.dst_md, concat_fwd_dims.concat_dims, context_.src_md, in Setup()
363 new memory(*context_.dst_md, cpu_engine_, DummyData)); in Setup()
664 memory::desc dst_md({}, memory::data_type::undef, in Compute() local
679 dst_md = MklDnnData<T>::CreateBlockedMemDesc( in Compute()
684 dst_md = memory::desc(dst_dims, MklDnnType<T>(), mkl_common_format); in Compute()
693 dst_md = memory::desc(dst_dims, MklDnnType<T>(), mkl_common_format); in Compute()
738 dst_md = dnn_shape_dst.GetMklLayout(); in Compute()
739 dst.SetUsrMem(dst_md, dst_tensor); in Compute()
[all …]
|
| D | mkl_matmul_op_fused.cc | 168 auto dst_md = in Compute() local
178 add_md = dst_md = in Compute()
184 auto fuse_add_dst_ = memory(dst_md, this->cpu_engine_, dst_buf); in Compute()
186 ReorderPd(this->cpu_engine_, add_md, this->cpu_engine_, dst_md); in Compute()
|
| D | mkl_pooling_ops_common.cc | 47 context_.dst_md.reset(new memory::desc({fwdParams.dst_dims}, MklDnnType<T>(), in Setup()
55 *context_.dst_md, fwdParams.strides, fwdParams.filter_dims, in Setup()
141 context_.dst_md.reset(new memory::desc({bwdParams.dst_dims}, MklDnnType<T>(), in Setup()
150 bwdParams.alg_kind, *context_.src_md, *context_.dst_md, bwdParams.strides, in Setup()
156 *context_.dst_md, bwdParams.strides, bwdParams.filter_dims, in Setup()
|
| D | mkl_conv_ops.cc | 187 std::shared_ptr<mkldnn::memory::desc> dst_md; member
224 context_.dst_md.reset(new memory::desc( in Setup()
237 *context_.dst_md, convFwdDims.strides, convFwdDims.dilations, in Setup()
242 *context_.src_md, *context_.filter_md, *context_.dst_md, in Setup()
855 auto dst_md = conv_prim_desc.dst_desc(); in AllocateOutputTensor() local
858 dst_md.data.data_type = in AllocateOutputTensor()
864 output_mkl_shape->SetMklLayout(&dst_md); in AllocateOutputTensor()
871 output_tf_shape.AddDim((dst_md.get_size() / sizeof(Toutput))); in AllocateOutputTensor()
914 add_md = dst_md = in AllocateOutputTensor()
919 fuse_add_dst_.reset(new memory(dst_md, this->cpu_engine_, dst_buf)); in AllocateOutputTensor()
[all …]
|
| D | mkl_pooling_ops_common.h | 127 std::shared_ptr<mkldnn::memory::desc> dst_md; member
146 dst_md(nullptr), in PoolingFwdContext()
264 std::shared_ptr<mkldnn::memory::desc> dst_md; member
287 dst_md(nullptr), in PoolingBwdContext()
|
| D | mkl_matmul_ops_common.h | 154 std::shared_ptr<mkldnn::memory::desc> dst_md; member
172 dst_md(nullptr), in MklDnnMatMulFwdContext()
187 context_.dst_md.reset(new memory::desc({matmul_fwd_params.dst_dims}, in Setup()
197 *context_.bias_md, *context_.dst_md)); in Setup()
|
| D | mkl_relu_op.cc | 112 std::shared_ptr<memory::desc> dst_md; member
130 dst_md(nullptr), in EltwiseFwdContext()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 32 #define DST_MD dst_md
111 #define REORDER_PD_CONSTRUCTOR(src_md, dst_md, engine) \ argument
112 ReorderPd(engine, src_md, engine, dst_md)
113 #define REORDER_PD_CONSTRUCTOR_WITH_ATTR(src_md, dst_md, engine, prim_attr) \ argument
114 ReorderPd(engine, src_md, engine, dst_md, prim_attr)
|