提交 7a1ddf26 编写于 作者: S Sang Han 提交者: Vijay Vasudevan

Fix casting to size_t for mkl conv filter dims (#13831)

上级 bbf1985d
......@@ -288,8 +288,10 @@ class MklConv2DOp : public OpKernel {
mkl_filter_output_mkl_shape.SetMklLayout(mkl_context.prim_fwd,
dnnResourceFilter);
size_t filter_sizes[4] = {filter.dim_size(0), filter.dim_size(1),
filter.dim_size(2), filter.dim_size(3)};
size_t filter_sizes[4] = {static_cast<size_t>(filter.dim_size(0)),
static_cast<size_t>(filter.dim_size(1)),
static_cast<size_t>(filter.dim_size(2)),
static_cast<size_t>(filter.dim_size(3))};
mkl_filter_output_mkl_shape.SetTfLayout(filter.dims(), filter_sizes,
mkl_context.filter_strides);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册