Home
last modified time | relevance | path

Searched refs:num_blocks_horizontally (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/image/kernels/
Dsegmentation_ops.cc95 int64 num_blocks_horizontally = union_find.num_blocks_horizontally(); in operator ()() local
103 num_images * num_blocks_vertically * num_blocks_horizontally, cost, in operator ()()
104 [&union_find, num_blocks_vertically, num_blocks_horizontally]( in operator ()()
107 int64 block_x = i % num_blocks_horizontally; in operator ()()
109 (i / num_blocks_horizontally) % num_blocks_vertically; in operator ()()
111 i / (num_blocks_horizontally * num_blocks_vertically); in operator ()()
Dsegmentation_ops.h89 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE int64 num_blocks_horizontally() const { in num_blocks_horizontally() function
95 return num_blocks_vertically() * num_blocks_horizontally(); in num_blocks()