Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.cc185 std::shared_ptr<mkldnn::memory::desc> filter_md; member
625 auto filter_md = in Compute() local
1014 auto filter_md = conv_prim_desc.weights_desc(); in AllocateFilterOutputTensor() local
1093 MklDnnData<Tfilter>& filter, const memory::desc& filter_md, in CacheFilter()
1118 bool AreMemoryDescriptorsEqual(const memory::desc& filter_md, in AreMemoryDescriptorsEqual()
1136 const memory::desc& filter_md) in GetCachedFilter()
Dmkl_conv_grad_input_ops.cc160 std::shared_ptr<memory::desc> filter_md; member
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h62 #define IS_FILTER_REORDER_NEEDED(filter_md, op_pd, op) \ argument