Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

support FusedDepthwiseConv2dNative op #2446

Merged
merged 2 commits into from
Dec 2, 2019
Merged

Conversation

pyu10055
Copy link
Collaborator

@pyu10055 pyu10055 commented Nov 27, 2019

Added support for FusedDepthwiseConv2dNative in GraphModel execution.
Fixed an issue with the python converter code, where the numArgs attribute is set wrong for FusedDepthwiseConv2dNative op.

To see the logs from the Cloud Build CI, please join either our discussion or announcement mailing list.


This change is Reviewable

Copy link
Contributor

@dsmilkov dsmilkov left a comment

Choose a reason for hiding this comment

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

One small comment

Reviewed 6 of 6 files at r1.
Reviewable status: :shipit: complete! 1 of 1 approvals obtained (waiting on @annxingyuan, @dsmilkov, and @pyu10055)


tfjs-converter/src/operations/executors/convolution_executor.ts, line 86 at r1 (raw file):

          if (isBatchNorm) {
            throw new Error(
                'Fused Conv2d with FusedBatchNorm is not supported.');

FusedConv2d/FusedDepthwiseConv2D

@pyu10055 pyu10055 merged commit a714642 into master Dec 2, 2019
@pyu10055 pyu10055 deleted the fused_depthwise_conv2d branch December 2, 2019 21:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants