| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 151 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingOp() local
152 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingOp()
432 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOp() local
433 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingGradOp()
519 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOpCustomGPUKernel() local
520 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingGradOpCustomGPUKernel()
|
| D | maxpooling_op.cc | 364 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradOp() local
365 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingGradOp()
416 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute() local
417 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in Compute()
681 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradGradOp() local
682 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingGradGradOp()
729 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute() local
730 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in Compute()
1181 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingNoMaskOp() local
1182 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingNoMaskOp()
[all …]
|
| D | conv_ops_using_gemm.cc | 446 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DUsingGemmOp() local
449 context, stride_n == 1 && stride_c == 1, in Conv2DUsingGemmOp()
|
| D | depthwise_conv_op.cc | 284 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeOp() local
292 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeOp()
|
| D | depthwise_conv_grad_op.cc | 565 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropInputOp() local
573 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeBackpropInputOp()
1068 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropFilterOp() local
1076 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeBackpropFilterOp()
|
| D | conv_ops_fused_image_transform.cc | 637 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); in FusedResizeConv2DUsingGemmOp() local
640 context, stride_n == 1 && stride_c == 1, in FusedResizeConv2DUsingGemmOp()
|
| D | conv_grad_input_ops.h | 366 int stride_n = GetTensorDim(strides_, data_format_, 'N');
371 context, (stride_n == 1 && stride_c == 1),
|
| D | conv_grad_filter_ops.cc | 270 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DBackpropFilterOp() local
275 context, (stride_n == 1 && stride_c == 1), in Conv2DBackpropFilterOp()
|
| D | conv_ops.cc | 371 const int64 stride_n = GetTensorDim(strides, data_format, 'N'); in InitConv2DParameters() local
376 stride_n == 1 && stride_c == 1, in InitConv2DParameters()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 569 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvBackpropCommonOp() local
572 context, (stride_n == 1 && stride_c == 1), in MklConvBackpropCommonOp()
|
| D | mkl_conv_ops.cc | 449 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp() local
452 context, stride_n == 1 && stride_c == 1, in MklConvOp()
|