Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.cc236 std::shared_ptr<dnnl::memory::desc> filter_md; member
763 auto filter_md = in Compute() local
1259 auto filter_md = conv_prim_desc.weights_desc(); in AllocateFilterOutputTensor() local
1297 MklDnnData<Tfilter>& filter, const memory::desc& filter_md, in CacheFilter()
1321 bool AreMemoryDescriptorsEqual(const memory::desc& filter_md, in AreMemoryDescriptorsEqual()
1339 const memory::desc& filter_md) in GetCachedFilter()
Dmkl_conv_grad_input_ops.cc150 std::shared_ptr<memory::desc> filter_md; member