Searched defs:weight_md (Results 1 – 3 of 3) sorted by relevance
238 const memory::desc weight_md = in Compute() local
183 std::shared_ptr<dnnl::memory::desc> weight_md; member504 MklDnnData<Tweight>& weight, const memory::desc& weight_md) in CacheWeight()
218 auto weight_md = weight_mkl_shape.IsMklTensor() in Compute() local