Home | History | Annotate | Download | only in kernels

Lines Matching refs:MklDnnShape

719       MklDnnShape dnn_shape_src;
868       MklDnnShape dnn_shape_dst;
955     MklDnnShape dnn_shape_dst;
988     MklDnnShape mkl_shape_batch_mean;
998     MklDnnShape mkl_shape_batch_variance;
1010     MklDnnShape mkl_shape_saved_mean;
1019     MklDnnShape mkl_shape_saved_variance;
1061       MklDnnShape dnn_shape_src, dnn_shape_diff_dst;
1212       MklDnnShape dnn_shape_diff_src;
1296     MklDnnShape dnn_shape_diff_src;
1322     MklDnnShape mkl_shape_diff_scale;
1330     MklDnnShape mkl_shape_diff_shift;
1341     MklDnnShape mkl_shape_p;