Searched refs:ksize_c (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes.py | 389 ksize_b, ksize_d, ksize_r, ksize_c = op.get_attr("ksize")
392 ksize_b, ksize_r, ksize_c, ksize_d = op.get_attr("ksize")
413 ksize_c, stride_r, stride_c,
457 ksize_b, ksize_d, ksize_r, ksize_c = op.get_attr("ksize")
460 ksize_b, ksize_r, ksize_c, ksize_d = op.get_attr("ksize")
475 if not ((ksize_r == 1 and ksize_c == 1) or ksize_d == 1):
485 ksize_c, stride_r, stride_c,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 762 _, ksize_r, ksize_c, _ = op.get_attr("ksizes")
764 output_indices_num = rows_out * cols_out * ksize_r * ksize_c
767 (1, rows_out, cols_out, ksize_r * ksize_c))
788 grad, (batch_size, rows_out, cols_out, ksize_r, ksize_c, channels)),
824 _, ksize_p, ksize_r, ksize_c, _ = op.get_attr("ksizes")
827 output_indices_num = prc_indices_num * ksize_p * ksize_r * ksize_c
830 (1, planes_out, rows_out, cols_out, ksize_p * ksize_r * ksize_c))
849 ksize_p, ksize_r, ksize_c, channels)),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 676 const int32 ksize_c = GetTensorDim(ksize_, data_format_, 'C'); in MaxPooling3dGradGradOp() local
678 OP_REQUIRES(context, ksize_c == 1 && stride_c == 1, in MaxPooling3dGradGradOp()
|