Searched defs:ksizes (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/image_ops/ |
| H A D | extract_volume_patches_grad_test.py | 56 def testGradient(self, in_shape, ksizes, strides): argument
71 def extract(in_val, ksizes=ksizes, strides=strides, padding=padding): argument
|
| H A D | extract_volume_patches_op_test.py | 26 def _VerifyValues(self, image, ksizes, strides, padding, patches): argument
|
| H A D | extract_image_patches_op_test.py | 27 def _VerifyValues(self, image, ksizes, strides, rates, padding, patches): argument
|
| H A D | extract_image_patches_grad_test.py | 101 ksizes=ksizes, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns_test.cc | 1127 const string& data_format) { in TEST()
1173 const string& data_format) { in TEST()
1209 const string& data_format) { in TEST()
1248 const std::vector<int32>& ksizes, const string& padding) { in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | extract_image_patches_op_test.py | 28 def _VerifyValues(self, image, ksizes, strides, rates, padding, patches): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops.cc | 2621 std::vector<int32> ksizes; in __anon38bbb0e83902() local
2711 std::vector<int32> ksizes; in __anon38bbb0e83a02() local
|
| H A D | array_ops_test.cc | 1324 const std::vector<int32>& rates, const string& padding) { in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 2609 llvm::ArrayRef<int64_t> ksizes, in replaceWithAvgPool()
2688 llvm::ArrayRef<int64_t> ksizes, in replaceWithMaxPool()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | array_ops.py | 6605 ksizes=None, argument
|