Skip to content

Add ops for 3d connected components - #9

Merged
yangliupku merged 1 commit into
masterfrom
bladder_cc_ops
Sep 19, 2019
Merged

yangliupku merged 1 commit into
masterfrom
bladder_cc_ops

Conversation

@yangliupku

Copy link
Copy Markdown

register ops:

  • Greater for int32
  • GreaterEqual for int 32
  • Bincount for int32

namespace tensorflow {
REGISTER9(BinaryOp, CPU, "Greater", functor::greater, float, Eigen::half,
double, int32, int64, uint8, int8, int16, bfloat16);
REGISTER(BinaryOp, CPU, "Greater", functor::greater_equal, int32);

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why int32 ?

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It's used by the tf.math.bincount function..

@yangliupku
yangliupku merged commit 89f0705 into master Sep 19, 2019
@AlonDaks
AlonDaks deleted the bladder_cc_ops branch September 19, 2019 18:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants