Searched refs:union_find (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| D | segmentation_ops.cc | 90 BlockedImageUnionFindFunctor<T> union_find( in operator ()() local
92 while (union_find.can_merge()) { in operator ()()
93 union_find.merge_blocks(); in operator ()()
94 int64 num_blocks_vertically = union_find.num_blocks_vertically(); in operator ()()
95 int64 num_blocks_horizontally = union_find.num_blocks_horizontally(); in operator ()()
101 int cost = (union_find.block_height() + union_find.block_width()) * 20; in operator ()()
104 [&union_find, num_blocks_vertically, num_blocks_horizontally]( in operator ()()
112 union_find.merge_internal_block_edges(image, block_y, block_x); in operator ()()
117 images.data(), union_find); in operator ()()
|
| D | segmentation_ops.h | 270 const BlockedImageUnionFindFunctor<T>& union_find) { in operator()
272 component_ids.generate(FindRootGenerator(images, union_find)); in operator()
282 const T* images, BlockedImageUnionFindFunctor<T> union_find) in FindRootGenerator() argument
283 : images_(images), union_find_(union_find) {} in FindRootGenerator()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 521 ":union_find",
581 name = "union_find",
582 hdrs = ["union_find.h"],
719 ":union_find",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | BUILD | 480 "//tensorflow/compiler/jit:union_find",
507 "//tensorflow/compiler/jit:union_find",
543 "//tensorflow/compiler/jit:union_find",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | BUILD | 352 "segment/union_find.h",
|