Searched refs:dst_md (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_concat_op.cc | 349 memory::desc dst_md({}, memory::data_undef, memory::format_undef); in Compute() local
359 dst_md = in Compute()
364 dst_md = memory::desc(dst_dims, MklDnnType<T>(), memory::format::nchw); in Compute()
413 dst_md = in Compute()
414 dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.GetMklLayout() : dst_md; in Compute()
415 dst.SetUsrMem(dst_md, dst_tensor); in Compute()
|
| D | mkl_conv_ops.cc | 189 std::shared_ptr<mkldnn::memory::desc> dst_md; member
222 context_.dst_md.reset(new memory::desc( in Setup()
233 *context_.filter_md, *context_.bias_md, *context_.dst_md, in Setup()
239 *context_.filter_md, *context_.dst_md, convFwdDims.strides, in Setup()
1257 auto dst_md = dst_pd.desc(); in AllocateOutputTensor() local
1259 dst_md.data.data_type = in AllocateOutputTensor()
1261 dst_pd = memory::primitive_desc(dst_md, cpu_engine_); in AllocateOutputTensor()
1771 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor() local
1776 dst_md.data.data_type = in AllocateOutputTensor()
1778 summand_mkl_shape.SetMklLayout(&dst_md); in AllocateOutputTensor()
|
| D | mkl_pooling_ops_common.cc | 59 context_.dst_md.reset(new memory::desc({fwdParams.dst_dims}, MklDnnType<T>(), in Setup()
65 *context_.dst_md, fwdParams.strides, fwdParams.filter_dims, in Setup()
|
| D | mkl_pooling_ops_common.h | 126 std::shared_ptr<mkldnn::memory::desc> dst_md; member
143 dst_md(nullptr), in PoolingFwdContext()
|
| D | mkl_relu_op.cc | 112 std::shared_ptr<memory::desc> dst_md; member
130 dst_md(nullptr), in EltwiseFwdContext()
|