Searched refs:mkldnn_matmul_fwd_dims (Results 1 – 1 of 1) sorted by relevance
295 const MklDnnMatMulFwdParams& mkldnn_matmul_fwd_dims, bool do_not_cache) { in Get() argument303 mkldnn_matmul_fwd_dims); in Get()310 .GetMklDnnMatMulFwd(mkldnn_matmul_fwd_dims)); in Get()314 mkldnn_matmul_fwd_dims); in Get()317 .SetMklDnnMatMulFwd(mkldnn_matmul_fwd_dims, matmul_fwd); in Get()332 static string CreateKey(const MklDnnMatMulFwdParams& mkldnn_matmul_fwd_dims) { in CreateKey() argument336 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.src_dims); in CreateKey()337 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.weight_dims); in CreateKey()338 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.bias_dims); in CreateKey()339 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.dst_dims); in CreateKey()[all …]