Home
last modified time | relevance | path

Searched refs:GetWindowedOutputSize (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc115 GetWindowedOutputSize(in_planes, ksize_planes, stride_planes, in Compute()
118 GetWindowedOutputSize(in_rows, ksize_rows, stride_rows, in Compute()
121 GetWindowedOutputSize(in_cols, ksize_cols, stride_cols, in Compute()
Dextract_image_patches_op.cc92 GetWindowedOutputSize(in_rows, ksize_rows_eff, stride_rows, in Compute()
95 GetWindowedOutputSize(in_cols, ksize_cols_eff, stride_cols, in Compute()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconv_grad_size_util.cc24 StatusOr<SpatialDimensionOutputSizeAndPadding> GetWindowedOutputSize( in GetWindowedOutputSize() function
71 GetWindowedOutputSize(input_size, filter_size, dilation, in ConvGradExtractAndVerifyDimension()
/external/tensorflow/tensorflow/core/framework/
Dkernel_shape_util.cc76 Status GetWindowedOutputSize(int64 input_size, int64 filter_size, int64 stride, in GetWindowedOutputSize() function
111 TF_RETURN_IF_ERROR(GetWindowedOutputSize(input[i], window[i], strides[i], in Get3dOutputSize()
Dkernel_shape_util.h74 Status GetWindowedOutputSize(int64 input_size, int64 filter_size, int64 stride,
/external/tensorflow/tensorflow/core/kernels/
Dops_util_test.cc74 Status status = GetWindowedOutputSize( in VerifyGet2dOutputSizeBoundaries()
79 status = GetWindowedOutputSize( in VerifyGet2dOutputSizeBoundaries()
89 Status status = GetWindowedOutputSize( in VerifyGet2dOutputSizeValues()
94 status = GetWindowedOutputSize( in VerifyGet2dOutputSizeValues()
Davgpooling_op.cc325 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
328 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
585 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
588 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
Dconv_ops_using_gemm.cc524 GetWindowedOutputSize(input_rows, filter_rows, stride_rows, in Compute()
527 GetWindowedOutputSize(input_cols, filter_cols, stride_cols, in Compute()
Dnn_ops_test.cc129 TF_CHECK_OK(GetWindowedOutputSize(rows, filter_rows, stride, padding, in BM_ConvFloat()
131 TF_CHECK_OK(GetWindowedOutputSize(cols, filter_cols, stride, padding, in BM_ConvFloat()
538 TF_CHECK_OK(GetWindowedOutputSize(rows, filter_rows, stride, padding, in BM_ConvFloatDepthwise()
540 TF_CHECK_OK(GetWindowedOutputSize(cols, filter_cols, stride, padding, in BM_ConvFloatDepthwise()
942 TF_CHECK_OK(GetWindowedOutputSize(rows, kernel_rows, stride, padding, in BM_AvgPoolBk()
944 TF_CHECK_OK(GetWindowedOutputSize(cols, kernel_cols, stride, padding, in BM_AvgPoolBk()
1130 TF_CHECK_OK(GetWindowedOutputSize(rows, kernel_rows, stride, padding, in BM_MaxPoolBk()
1132 TF_CHECK_OK(GetWindowedOutputSize(cols, kernel_cols, stride, padding, in BM_MaxPoolBk()
Dquantized_conv_ops.cc542 GetWindowedOutputSize(input_rows, filter_rows, stride, in Compute()
545 GetWindowedOutputSize(input_cols, filter_cols, stride, in Compute()
Ddilation_ops.cc107 context, GetWindowedOutputSize(input_rows, filter_rows_eff, *stride_rows, in ParseSizes()
110 context, GetWindowedOutputSize(input_cols, filter_cols_eff, *stride_cols, in ParseSizes()
Dpooling_ops_3d.cc79 OP_REQUIRES_OK(context, GetWindowedOutputSize(tensor_in_planes, window_planes, in Pool3dParameters()
83 GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride, in Pool3dParameters()
86 GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in Pool3dParameters()
Dconv_ops_fused_image_transform.cc824 GetWindowedOutputSize(padded_rows, filter_rows, stride_rows, in Compute()
827 GetWindowedOutputSize(padded_cols, filter_cols, stride_cols, in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_window_utils.h37 DynamicWindowDims GetWindowedOutputSize(HloInstruction* input_size,
Ddynamic_window_utils.cc100 DynamicWindowDims GetWindowedOutputSize(HloInstruction* input_size, in GetWindowedOutputSize() function
139 DynamicWindowDims forward_dims = GetWindowedOutputSize( in GetWindowedInputGradSize()
Ddynamic_padder.cc900 DynamicWindowDims dynamic_window_dims = GetWindowedOutputSize( in RewriteDynamicConvolutionForward()
978 DynamicWindowDims dynamic_window_dims = GetWindowedOutputSize( in RewriteDynamicConvolutionKernelGrad()
1044 DynamicWindowDims dynamic_window_dims = GetWindowedOutputSize( in RewriteDynamicReduceWindowSamePadding()
1099 DynamicWindowDims dynamic_window_dims = GetWindowedOutputSize( in RewriteDynamicSelectAndScatterSamePadding()
Ddynamic_dimension_inference.cc714 DynamicWindowDims dynamic_window_dims = GetWindowedOutputSize( in HandleDynamicConvolutionForward()
733 DynamicWindowDims dynamic_window_dims = GetWindowedOutputSize( in HandleDynamicWindowSamePadding()
1068 DynamicWindowDims dynamic_window_dims = GetWindowedOutputSize( in HandleReduceWindow()
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc92 GetWindowedOutputSize(input_rows, filter_rows, stride, in Compute()
95 GetWindowedOutputSize(input_cols, filter_cols, stride, in Compute()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc667 s = GetWindowedOutputSize(d.input_dims[i], d.kernel_dims[i], in ChooseWindowedSpatialDims()