Lines Matching refs:in_sizes
135 mkl_context.mkl_params.in_sizes,
198 size_t orig_size = mkl_context.mkl_params.in_sizes[0] *
199 mkl_context.mkl_params.in_sizes[1] *
200 mkl_context.mkl_params.in_sizes[3];
220 size_t in_sizes[4];
247 mkl_params.in_sizes[0] = static_cast<size_t>(
250 mkl_params.in_sizes[1] = static_cast<size_t>(
253 mkl_params.in_sizes[2] = static_cast<size_t>(
256 mkl_params.in_sizes[3] = static_cast<size_t>(
259 mkl_params.depth = mkl_params.in_sizes[2];
261 mkl_params.in_sizes);
273 mkl_params.in_sizes, mkl_params.in_strides),
422 mkl_context.mkl_params.in_sizes,
498 size_t in_sizes[4];
530 mkl_params.in_sizes[0] = static_cast<size_t>(
533 mkl_params.in_sizes[1] = static_cast<size_t>(
536 mkl_params.in_sizes[2] = static_cast<size_t>(
539 mkl_params.in_sizes[3] = static_cast<size_t>(
542 mkl_params.depth = mkl_params.in_sizes[2];
544 mkl_params.in_sizes);
555 mkl_params.in_sizes, mkl_params.in_strides),
566 mkl_params.in_sizes, mkl_params.in_strides),