Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_ops_common.h46 struct MklDnnMatMulFwdParams { struct
61 MklDnnMatMulFwdParams(memory::dims src_dims, memory::dims weight_dims, argument
86 const MklDnnMatMulFwdParams& matmulFwdParams) in MklDnnMatMulFwdPrimitive() argument
176 void Setup(const MklDnnMatMulFwdParams& matmul_fwd_params) { in Setup()
295 const MklDnnMatMulFwdParams& mkldnn_matmul_fwd_dims, bool do_not_cache) { in Get()
332 static string CreateKey(const MklDnnMatMulFwdParams& mkldnn_matmul_fwd_dims) { in CreateKey()
369 const MklDnnMatMulFwdParams& mkldnn_matmul_fwd_dims) { in GetMklDnnMatMulFwd()
374 void SetMklDnnMatMulFwd(const MklDnnMatMulFwdParams& mkldnn_matmul_fwd_dims, in SetMklDnnMatMulFwd()
Dmkl_matmul_op_fused.cc119 MklDnnMatMulFwdParams matmul_params( in Compute()
266 MklDnnMatMulFwdParams& params) { in ExtendMklDnnMatMulFwdParams()
Dmkl_qmatmul_op.cc227 MklDnnMatMulFwdParams matmul_fwd_dims(src_dims, weight_dims, bias_dims, in Compute()
350 MklDnnMatMulFwdParams& params) { in ExtendMklDnnMatMulFwdParams()
523 MklDnnMatMulFwdParams& params) override { in ExtendMklDnnMatMulFwdParams()