Lines Matching refs:MklDnnShape
515 MklDnnShape dnn_shape_input;
595 MklDnnShape workspace_mkl_shape;
630 MklDnnShape orig_input_mkl_shape, orig_output_mkl_shape, grad_mkl_shape,
724 const MklDnnShape& original_input_mkl_shape,
746 const MklDnnShape& orig_input_mkl_shape,
747 const MklDnnShape& orig_output_mkl_shape,
748 const MklDnnShape& grad_mkl_shape,
749 const MklDnnShape& workspace_mkl_shape) {
