Home
last modified time | relevance | path

Searched defs:ksize (Results 1 – 25 of 41) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dpooling_ops_test.py70 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
106 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
295 ksize, argument
446 ksize, argument
550 def AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding, argument
H A Dpooling_ops_3d_test.py30 def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding): argument
190 ksize, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py4467 def avg_pool_v2(input, ksize, strides, padding, data_format=None, name=None): # pylint: disable=re… argument
4534 def avg_pool(value, ksize, strides, padding, data_format="NHWC", argument
4579 def avg_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): # pylint: disable=r… argument
4620 def avg_pool1d(input, ksize, strides, padding, data_format="NWC", name=None): # pylint: disable=re… argument
4667 def avg_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): # pylint: disable=… argument
4709 def max_pool_v2(input, ksize, strides, padding, data_format=None, name=None): argument
4849 ksize, argument
4910 def max_pool1d(input, ksize, strides, padding, data_format="NWC", name=None): argument
4970 def max_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): argument
5080 def max_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): argument
[all …]
H A Dinit_ops.py1220 def _orthogonal_kernel(self, ksize, cin, cout): argument
1352 def _orthogonal_kernel(self, ksize, cin, cout): argument
1517 def _orthogonal_kernel(self, ksize, cin, cout): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_pooling_ops_common.cc230 const std::vector<int32>& ksize, in Init()
257 const std::vector<int32>& ksize, in Init()
282 const std::vector<int32>& ksize, in Init()
H A Dmkl_quantized_pooling_ops_test.cc37 const int ksize = 2; in TEST_F() local
94 const int ksize = 2; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dnn_grad_test.cc253 const std::vector<int> ksize{1, 2, 2, 1}; in TEST_F() local
266 Tensor ksize = test::AsTensor<int>({1, 2, 2, 1}, {4}); in TEST_F() local
279 const std::vector<int> ksize{1, 3, 3, 3, 1}; in TEST_F() local
292 const std::vector<int> ksize{1, 2, 2, 1}; in TEST_F() local
303 const std::vector<int> ksize{1, 3, 3, 3, 1}; in TEST_F() local
H A Dnn_grad.cc246 std::vector<int32> ksize; in MaxPoolGradHelper() local
281 std::vector<int32> ksize; in MaxPool3DGradHelper() local
302 std::vector<int32> ksize; in AvgPoolGradHelper() local
323 std::vector<int32> ksize; in AvgPool3DGradHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dpooling_ops_test.py187 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
281 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
327 ksize, argument
875 ksize, argument
906 def _CompareMaxPoolingFwd(self, input_shape, ksize, strides, padding): argument
922 def _CompareMaxPoolingBk(self, input_shape, output_shape, ksize, strides, argument
955 def _CompareMaxPoolingGradBk(self, input_shape, output_shape, ksize, strides, argument
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_image_patch.cpp189 int ksize = 2; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid() local
265 int ksize = 3; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid_same_value() local
338 int ksize = 2; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_same() local
415 int ksize = 1; // Corresponds to the Rows and Cols for in test_patch_padding_same_negative_padding_clip_to_zero() local
H A Dcxx11_tensor_image_patch_sycl.cpp264 IndexType ksize = 2; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid_sycl() local
370 IndexType ksize = 3; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid_same_value_sycl() local
467 IndexType ksize = 2; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_same_sycl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op.cc290 std::vector<int32> ksize = ksize_; in Compute() local
408 std::vector<int32> ksize = ksize_; in Compute() local
514 std::vector<int32> ksize = ksize_; in Compute() local
732 std::vector<int32> ksize = ksize_; in Compute() local
895 std::vector<int32> ksize = ksize_; in Compute() local
1365 std::vector<int32> ksize = ksize_; in Compute() local
H A Dquantized_pooling_ops_test.cc39 const int ksize = 2; in TEST_F() local
84 const int ksize = 2; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc101 std::vector<int64_t> ksize; in GetKernelSize() local
179 std::vector<int64_t> ksize = ksize_or_error.ValueOrDie(); in Compile() local
268 std::vector<int64_t> ksize = ksize_or_error.ValueOrDie(); in Compile() local
/aosp_15_r20/external/eigen/bench/
H A Dbenchmark-blocking-sizes.cpp572 for (size_t ksize = minsize; ksize <= maxsize; ksize *= 2) { in run() local
603 for (size_t ksize = minsize; ksize <= maxsize; ksize *= 2) { in run() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dops_util.cc40 Status GetBroadcastSize(const int index, const int in_size, const int ksize, in GetBroadcastSize()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dpool.py115 def make_l2_pool(input_tensor, ksize, strides, padding, data_format): argument
/aosp_15_r20/external/linux-kselftest/tools/testing/selftests/bpf/progs/
H A Dtest_get_stack_rawtp.c62 long usize, ksize; in bpf_prog1() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dpool_test.cc165 const int ksize = 15; in TEST() local
/aosp_15_r20/external/libopus/dnn/
H A Dnnet.c135 int ksize = layer->nb_inputs/input_size; in compute_generic_conv1d_dilation() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleKernelAVXAntialias.h160 int ksize, in ImagingResampleHorizontal()
238 int ksize, in ImagingResampleVertical()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator_test.cc360 const std::vector<int>& ksize, in GetPoolingOutputSize()
430 const std::vector<int>& ksize, in DescribePoolingOp()
594 const std::vector<int> ksize = {1, kx, ky, 1}; in CallOpDimensionsFromInputs() local
/aosp_15_r20/cts/apps/CtsVerifier/libs/
HDopencv3-android.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/opencv/ org/ ...
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dpooling.cc31 absl::Span<const int64_t> ksize, absl::Span<const int64_t> stride, in AvgPoolDivideByCountWithGeneralPadding()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc1001 auto& ksize = (*maxpool_op->mutable_attr())["ksize"]; in ConvertMaxPoolOperator() local
1029 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertAveragePoolOperator() local
1106 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertL2PoolOperator() local

12