Searched refs:c_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 514 memory::dims c_dims; member
519 MklMatMulParams(memory::dims a_dims, memory::dims b_dims, memory::dims c_dims, in MklMatMulParams()
524 c_dims(c_dims), in MklMatMulParams()
606 new memory::desc({params.c_dims}, MklDnnType<T>(), params.c_strides)); in Setup()
674 key_creator.AddAsKey(params.c_dims); in CreateKey()
704 dims c_dims = dims{m, n};
714 MklMatMulParams params(a_dims, b_dims, c_dims, a_strides, b_strides,
|