Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op_fused.cc235 this->GetCachedWeight(ctx, matmul_pd->weights_desc()); in Compute()
Dmkl_qmatmul_op.cc274 this->GetCachedWeight(context, matmul_fwd_pd->weights_desc()); in Compute()
Dmkl_matmul_ops_common.h471 Tweight* GetCachedWeight(OpKernelContext* context, in GetCachedWeight() function