Searched refs:ksize_n (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 134 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingOp() local
136 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingOp()
394 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingGradOp() local
396 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingGradOp()
474 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingGradOpCustomGPUKernel() local
476 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingGradOpCustomGPUKernel()
|
| D | maxpooling_op.cc | 365 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in MaxPoolingGradOp() local
367 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingGradOp()
413 const int32 ksize_n = GetTensorDim(ksize, data_format_, 'N'); in Compute() local
415 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in Compute()
672 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in MaxPoolingGradGradOp() local
674 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingGradGradOp()
720 const int32 ksize_n = GetTensorDim(ksize, data_format_, 'N'); in Compute() local
722 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in Compute()
1135 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in MaxPoolingNoMaskOp() local
1137 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingNoMaskOp()
[all …]
|