Home
last modified time | relevance | path

Searched defs:stride_n (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op.cc135 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingOp() local
395 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOp() local
475 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOpCustomGPUKernel() local
Dmaxpooling_op.cc366 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradOp() local
414 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute() local
673 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradGradOp() local
721 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute() local
1136 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingNoMaskOp() local
1227 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingNoMaskV2Op() local
1262 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute() local
Ddepthwise_conv_op.cc279 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeOp() local
Dmkl_conv_ops.h560 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvBackpropCommonOp() local
Ddepthwise_conv_grad_op.cc553 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropInputOp() local
1027 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropFilterOp() local
Dconv_ops_using_gemm.cc444 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DUsingGemmOp() local
Dconv_grad_filter_ops.cc453 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DSlowBackpropFilterOp() local
Dmkl_conv_ops.cc431 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp() local
875 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp() local
Dconv_grad_input_ops.cc590 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DSlowBackpropInputOp() local
Dconv_ops_fused_image_transform.cc636 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); in FusedResizeConv2DUsingGemmOp() local
Dconv_ops.cc304 const int64 stride_n = GetTensorDim(strides, data_format, 'N'); in InitConv2DParameters() local
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1391 int64 stride_n = H * W * C; in MklNHWCToNCHW() local
1407 int64 stride_n = H * W * C; in MklNCHWToNHWC() local