Lines Matching defs:in_dims
421 mkl_shape_in_backprop.SetTfLayout(mkl_context.mkl_params.in_dims,
424 mkl_shape_in_backprop.SetTfDimOrder(mkl_context.mkl_params.in_dims,
497 size_t in_dims;
527 mkl_params.in_dims = input_in_mkl_format
554 dnnLayoutCreate_F32(&mkl_lt_input, mkl_params.in_dims,
565 dnnLayoutCreate_F32(&mkl_lt_out_backprop, mkl_params.in_dims,
