experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Verify filter dimensions sanity in Conv2DBackpropInput

Browse Source 
PiperOrigin-RevId: 321399463
Change-Id: Ic196445c1cb120168fff16acc44e1e5a080c5c5c
This commit is contained in:
Eugene Zhulenev 2020-07-15 11:16:07 -07:00 committed by TensorFlower Gardener
parent 9edc5097b9
commit ee781437e8
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/kernels/conv_grad_input_ops.cc
View File

@ -561,6 +561,16 @@ class Conv2DCustomBackpropInputOp : public OpKernel {
/*dilations=*/{1, 1, 1, 1}, strides_, padding_,
explicit_paddings_, data_format_, &dims));
OP_REQUIRES(context, dims.in_depth == filter.shape().dim_size(2),
errors::InvalidArgument("Computed input depth ", dims.in_depth,
" doesn't match filter input depth ",
filter.shape().dim_size(2)));
OP_REQUIRES(
context, dims.out_depth == filter.shape().dim_size(3),
errors::InvalidArgument("Computed output depth ", dims.out_depth,
" doesn't match filter output depth ",
filter.shape().dim_size(3)));
Tensor* in_backprop = nullptr;
OP_REQUIRES_OK(context,
context->allocate_output(0, input_shape, &in_backprop));