Home | History | Annotate | Download | only in kernels

Lines Matching refs:MklDnnShape

388       MklDnnShape dnn_shape_src;
423       MklDnnShape dnn_shape_dst;
487       MklDnnShape dnn_shape_src, dnn_shape_diff_dst;
521         // in Mkl format, we just get Mkl layout from MklDnnShape. For input in
580       MklDnnShape dnn_shape_diff_src;
643     MklDnnShape dnn_shape_src;
649     MklDnnShape dnn_shape_dst;
676     MklDnnShape dnn_shape_diff_dst;
679     MklDnnShape dnn_shape_diff_src;
706     MklDnnShape dnn_shape_src;
712     MklDnnShape dnn_shape_dst;
743     MklDnnShape dnn_shape_diff_dst;
746     MklDnnShape dnn_shape_diff_src;
778     MklDnnShape dnn_shape_src;
784     MklDnnShape dnn_shape_dst;
814     MklDnnShape dnn_shape_diff_dst;
817     MklDnnShape dnn_shape_diff_src;