Searched refs:mkl_pool_params (Results 1 – 1 of 1) sorted by relevance
495 void GetOutputDims(const MklPoolParameters& mkl_pool_params, in GetOutputDims() argument499 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()500 mkl_pool_params.out_depth, in GetOutputDims()501 static_cast<int>(mkl_pool_params.out_height), in GetOutputDims()502 static_cast<int>(mkl_pool_params.out_width)}; in GetOutputDims()505 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()506 mkl_pool_params.out_depth, in GetOutputDims()507 static_cast<int>(mkl_pool_params.out_planes), in GetOutputDims()508 static_cast<int>(mkl_pool_params.out_height), in GetOutputDims()509 static_cast<int>(mkl_pool_params.out_width)}; in GetOutputDims()