| /external/tensorflow/tensorflow/compiler/tests/ |
| D | extract_image_patches_op_test.py | 28 def _VerifyValues(self, image, ksizes, strides, rates, padding, patches): argument
39 ksizes = [1] + ksizes + [1]
48 ksizes=ksizes,
65 ksizes=[1, 1],
80 ksizes=[1, 1],
94 ksizes=[2, 2],
108 ksizes=[2, 2],
123 ksizes=[2, 2],
137 ksizes=[2, 2],
|
| /external/tensorflow/tensorflow/python/kernel_tests/image_ops/ |
| D | extract_image_patches_op_test.py | 27 def _VerifyValues(self, image, ksizes, strides, rates, padding, patches): argument
38 ksizes = [1] + ksizes + [1]
44 ksizes=ksizes,
60 ksizes=[1, 1],
75 ksizes=[1, 1],
89 ksizes=[2, 2],
103 ksizes=[2, 2],
118 ksizes=[2, 2],
136 ksizes=[1, 1],
|
| D | extract_volume_patches_op_test.py | 26 def _VerifyValues(self, image, ksizes, strides, padding, patches): argument
41 ksizes = [1] + ksizes + [1]
47 ksizes=ksizes,
61 ksizes=[1, 1, 1],
73 ksizes=[1, 1, 1],
92 ksizes=[1, 1, 2],
103 ksizes=[2, 2, 2],
121 ksizes=[2, 2, 2],
|
| 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
72 return array_ops.extract_volume_patches(in_val, ksizes, strides,
97 ksizes=[1, ksize, ksize, ksize, 1],
|
| D | extract_image_patches_grad_test.py | 94 ksizes = tuple(test_case['ksizes'])
101 ksizes=ksizes, argument
105 return array_ops.extract_image_patches(in_val, ksizes, strides,
125 ksizes=[1, ksize, ksize, 1],
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ExtractImagePatches.pbtxt | 12 name: "ksizes"
68 name: "ksizes"
126 name: "ksizes"
185 name: "ksizes"
244 name: "ksizes"
|
| D | ExtractVolumePatches.pbtxt | 12 name: "ksizes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ExtractImagePatches.pbtxt | 12 name: "ksizes"
68 name: "ksizes"
126 name: "ksizes"
185 name: "ksizes"
244 name: "ksizes"
|
| D | ExtractVolumePatches.pbtxt | 12 name: "ksizes"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractVolumePatches.pbtxt | 20 name: "ksizes"
40 ksizes = [1, ksize_planes, ksize_rows, ksize_cols, 1]
|
| D | api_def_ExtractImagePatches.pbtxt | 19 name: "ksizes"
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns_test.cc | 1126 const std::vector<int32>& ksizes, const string& padding, in TEST()
1131 .Attr("ksize", ksizes) in TEST()
1172 const std::vector<int32>& ksizes, const string& padding, in TEST()
1177 .Attr("ksize", ksizes) in TEST()
1208 const std::vector<int32>& ksizes, const string& padding, in TEST()
1217 ksizes_tensor = test::AsTensor<int32>(ksizes); in TEST()
1248 const std::vector<int32>& ksizes, const string& padding) { in TEST() argument
1252 .Attr("ksize", ksizes) in TEST()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2621 std::vector<int32> ksizes; in __anon091a71b73902() local
2622 TF_RETURN_IF_ERROR(c->GetAttr("ksizes", &ksizes)); in __anon091a71b73902()
2623 if (ksizes.size() != 4) { in __anon091a71b73902()
2627 ksizes.size()); in __anon091a71b73902()
2648 int32_t ksize_rows = ksizes[1]; in __anon091a71b73902()
2649 int32_t ksize_cols = ksizes[2]; in __anon091a71b73902()
2711 std::vector<int32> ksizes; in __anon091a71b73a02() local
2712 TF_RETURN_IF_ERROR(c->GetAttr("ksizes", &ksizes)); in __anon091a71b73a02()
2713 if (ksizes.size() != 5) { in __anon091a71b73a02()
2717 ksizes.size()); in __anon091a71b73a02()
[all …]
|
| D | array_ops_test.cc | 1322 auto set_op = [&op](const std::vector<int32>& ksizes, in TEST()
1327 .Attr("ksizes", ksizes) in TEST()
|
| D | ops.pbtxt | 18269 name: "ksizes"
18355 name: "ksizes"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 2609 llvm::ArrayRef<int64_t> ksizes, in replaceWithAvgPool() argument
2613 if (ksizes.size() == 4) { in replaceWithAvgPool()
2615 op, op.getType(), input, rewriter.getI64ArrayAttr(ksizes), in replaceWithAvgPool()
2619 } else if (ksizes.size() == 5) { in replaceWithAvgPool()
2621 op, op.getType(), input, rewriter.getI64ArrayAttr(ksizes), in replaceWithAvgPool()
2688 llvm::ArrayRef<int64_t> ksizes, in replaceWithMaxPool() argument
2692 if (ksizes.size() == 4) { in replaceWithMaxPool()
2694 op, op.getType(0), input, rewriter.getI64ArrayAttr(ksizes), in replaceWithMaxPool()
2699 } else if (ksizes.size() == 5) { in replaceWithMaxPool()
2701 op, op.getType(0), input, rewriter.getI64ArrayAttr(ksizes), in replaceWithMaxPool()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 6605 ksizes=None, argument
6639 ksizes = deprecation.deprecated_argument_lookup("sizes", sizes, "ksizes",
6640 ksizes)
6641 return gen_array_ops.extract_image_patches(images, ksizes, strides, rates,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.image.pbtxt | 93 …argspec: "args=[\'images\', \'ksizes\', \'strides\', \'rates\', \'padding\', \'name\', \'sizes\'],…
|
| D | tensorflow.pbtxt | 1269 …argspec: "args=[\'images\', \'ksizes\', \'strides\', \'rates\', \'padding\', \'name\', \'sizes\'],…
1273 …argspec: "args=[\'input\', \'ksizes\', \'strides\', \'padding\', \'name\'], varargs=None, keywords…
|
| D | tensorflow.raw_ops.pbtxt | 1645 …argspec: "args=[\'images\', \'ksizes\', \'strides\', \'rates\', \'padding\', \'name\'], varargs=No…
1653 …argspec: "args=[\'input\', \'ksizes\', \'strides\', \'padding\', \'name\'], varargs=None, keywords…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 653 …argspec: "args=[\'input\', \'ksizes\', \'strides\', \'padding\', \'name\'], varargs=None, keywords…
|
| D | tensorflow.raw_ops.pbtxt | 1645 …argspec: "args=[\'images\', \'ksizes\', \'strides\', \'rates\', \'padding\', \'name\'], varargs=No…
1653 …argspec: "args=[\'input\', \'ksizes\', \'strides\', \'padding\', \'name\'], varargs=None, keywords…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 15986 func ExtractImagePatches(scope *Scope, images tf.Output, ksizes []int64, strides []int64, rates []i…
15990 …attrs := map[string]interface{}{"ksizes": ksizes, "strides": strides, "rates": rates, "padding": p…
16068 func ExtractVolumePatches(scope *Scope, input tf.Output, ksizes []int64, strides []int64, padding s…
16072 attrs := map[string]interface{}{"ksizes": ksizes, "strides": strides, "padding": padding}
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 24738 name: "ksizes"
24794 name: "ksizes"
24852 name: "ksizes"
24911 name: "ksizes"
24994 name: "ksizes"
|
| D | ops_history.v0.pbtxt | 11805 name: "ksizes"
|