Searched defs:weight_md (Results 1 – 3 of 3) sorted by relevance
224 const memory::desc weight_md = in Compute() local
152 std::shared_ptr<mkldnn::memory::desc> weight_md; member428 MklDnnData<Tweight>& weight, const memory::desc& weight_md) in CacheWeight()
217 auto weight_md = weight_mkl_shape.IsMklTensor() in Compute() local