Home
last modified time | relevance | path

Searched refs:filter_md (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc160 std::shared_ptr<memory::desc> filter_md; member
177 filter_md(nullptr), in ConvBwdInputContext()
196 context_.filter_md.reset(new memory::desc({convBwdInputDims.filter_dims}, in Setup()
203 *context_.filter_md, *context_.diff_dst_md, convBwdInputDims.strides, in Setup()
209 *context_.diff_src_md, *context_.filter_md, *context_.diff_dst_md, in Setup()
Dmkl_conv_ops.cc185 std::shared_ptr<mkldnn::memory::desc> filter_md; member
203 filter_md(nullptr), in ConvFwdContext()
220 context_.filter_md.reset(new memory::desc({convFwdDims.filter_dims}, in Setup()
236 *context_.src_md, *context_.filter_md, *context_.bias_md, in Setup()
242 *context_.src_md, *context_.filter_md, *context_.dst_md, in Setup()
625 auto filter_md = in Compute() local
629 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
691 if (filter_md != conv_fwd_pd->weights_desc()) { in Compute()
701 filter, filter_md, filter_mkl_shape); in Compute()
707 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h62 #define IS_FILTER_REORDER_NEEDED(filter_md, op_pd, op) \ argument
63 filter_md != op_pd->weights_desc()