| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 135 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingOp() local
136 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingOp()
395 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOp() local
396 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingGradOp()
475 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOpCustomGPUKernel() local
476 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in AvgPoolingGradOpCustomGPUKernel()
|
| D | maxpooling_op.cc | 366 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradOp() local
367 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingGradOp()
414 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute() local
415 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in Compute()
673 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradGradOp() local
674 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingGradGradOp()
721 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute() local
722 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in Compute()
1136 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingNoMaskOp() local
1137 OP_REQUIRES(context, ksize_n == 1 && stride_n == 1, in MaxPoolingNoMaskOp()
[all …]
|
| D | conv_ops_using_gemm.cc | 444 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DUsingGemmOp() local
447 context, stride_n == 1 && stride_c == 1, in Conv2DUsingGemmOp()
|
| D | depthwise_conv_op.cc | 279 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeOp() local
287 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeOp()
|
| D | depthwise_conv_grad_op.cc | 553 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropInputOp() local
561 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeBackpropInputOp()
1027 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropFilterOp() local
1035 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeBackpropFilterOp()
|
| D | mkl_conv_ops.h | 560 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvBackpropCommonOp() local
565 context, (stride_n == 1 && stride_c == 1), in MklConvBackpropCommonOp()
|
| D | conv_ops_fused_image_transform.cc | 636 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); in FusedResizeConv2DUsingGemmOp() local
639 context, stride_n == 1 && stride_c == 1, in FusedResizeConv2DUsingGemmOp()
|
| D | mkl_conv_ops.cc | 431 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp() local
434 context, stride_n == 1 && stride_c == 1, in MklConvOp()
875 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp() local
878 context, stride_n == 1 && stride_c == 1, in MklConvOp()
|
| D | conv_ops.cc | 304 const int64 stride_n = GetTensorDim(strides, data_format, 'N'); in InitConv2DParameters() local
309 stride_n == 1 && stride_c == 1, in InitConv2DParameters()
|
| D | conv_grad_filter_ops.cc | 453 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DSlowBackpropFilterOp() local
458 context, (stride_n == 1 && stride_c == 1), in Conv2DSlowBackpropFilterOp()
|
| D | conv_grad_input_ops.cc | 590 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DSlowBackpropInputOp() local
595 context, (stride_n == 1 && stride_c == 1), in Conv2DSlowBackpropInputOp()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1391 int64 stride_n = H * W * C; in MklNHWCToNCHW() local
1394 mkl_somatcopy('R', 'T', H * W, C, 1, buf_in + n * stride_n, C, in MklNHWCToNCHW()
1395 buf_out + n * stride_n, H * W); in MklNHWCToNCHW()
1407 int64 stride_n = H * W * C; in MklNCHWToNHWC() local
1410 mkl_somatcopy('R', 'T', C, H * W, 1, buf_in + n * stride_n, H * W, in MklNCHWToNHWC()
1411 buf_out + n * stride_n, C); in MklNCHWToNHWC()
|