search

tensorflow / tensorflow

An Open Source Machine Learning Framework for Everyone
https://tensorflow.org
Apache License 2.0
182.91k stars 73.92k forks source link

gpu_delegate: Update GPU MUL compatibility checker #67132

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

gpu_delegate: Update GPU MUL compatibility checker

ADD, MUL only works when two inputs has the same number of dimension on GPU. Make sure if the condition is checked properly.