Home
last modified time | relevance | path

Searched refs:input_dimension (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc55 int64 input_dimension = input_dimensions[i]; in MakePadding() local
123 tensorflow::MathUtil::CeilOfRatio(input_dimension, window_stride); in MakePadding()
126 window_dimension - input_dimension, in MakePadding()
Dpadding_test.cc27 std::pair<int64, int64> ComputePadding(int64 input_dimension, in ComputePadding() argument
30 return MakePadding({input_dimension}, {window_dimension}, {window_stride}, in ComputePadding()
/external/XNNPACK/src/operators/
Dunpooling-nhwc.c25 size_t input_dimension, in compute_output_dimension() argument
29 return doz(kernel_dimension * input_dimension, input_padding_dimension); in compute_output_dimension()
Dmax-pooling-nhwc.c39 size_t input_dimension, in compute_output_dimension_with_tf_same_padding() argument
42 return divide_round_up(input_dimension, stride_dimension); in compute_output_dimension_with_tf_same_padding()
Daverage-pooling-nhwc.c37 size_t input_dimension, in compute_output_dimension_with_tf_same_padding() argument
40 return divide_round_up(input_dimension, stride_dimension); in compute_output_dimension_with_tf_same_padding()
Ddeconvolution-nhwc.c28 size_t input_dimension, in compute_output_dimension() argument
37 stride_dimension * (input_dimension - 1) + adjustment_dimension + effective_kernel_dimension, in compute_output_dimension()
Dconvolution-nhwc.c41 size_t input_dimension, in compute_output_dimension_with_tf_same_padding() argument
44 return divide_round_up(input_dimension, subsampling_dimension); in compute_output_dimension_with_tf_same_padding()