Searched refs:ksize_c (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 1012 _, ksize_r, ksize_c, _ = op.get_attr("ksizes")
1014 output_indices_num = rows_out * cols_out * ksize_r * ksize_c
1017 (1, rows_out, cols_out, ksize_r * ksize_c))
1037 (batch_size, rows_out, cols_out, ksize_r, ksize_c, channels)),
1073 _, ksize_p, ksize_r, ksize_c, _ = op.get_attr("ksizes")
1076 output_indices_num = prc_indices_num * ksize_p * ksize_r * ksize_c
1079 (1, planes_out, rows_out, cols_out, ksize_p * ksize_r * ksize_c))
1100 ksize_c, channels)), (1, 2, 3, 4, 5, 6, 0, 7))
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 672 const int32 ksize_c = GetTensorDim(ksize_, data_format_, 'C'); in MaxPooling3dGradGradOp() local
674 OP_REQUIRES(context, ksize_c == 1 && stride_c == 1, in MaxPooling3dGradGradOp()
|