Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_ops_common.h476 const Tensor& weight_md_t = *weight_oi_md_.AccessTensor(context); in GetCachedWeight() local
480 if (weight_md_t.flat<Tweight>().size()) { in GetCachedWeight()
482 *(static_cast<memory::desc*>(weight_md_t.data())); in GetCachedWeight()