OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

1. Enable support for kDivide and kMaximum in XnnGraphFusion.

Browse Source 
2. Extend the test coverage.

PiperOrigin-RevId: 774945991
This commit is contained in:
A. Unique TensorFlower 2025-06-23 15:18:21 -07:00 committed by TensorFlower Gardener
parent bbd497fcda
commit fab3235983
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
third_party/xla/xla/backends/cpu/xnn_fusion.cc vendored
View File

@ -205,11 +205,10 @@ absl::StatusOr<xnn_binary_operator> XnnBinaryOperator(const HloOpcode& opcode) {
return xnn_binary_add;
case HloOpcode::kAnd:
return xnn_binary_bitwise_and;
// TODO(ashaposhnikov): debug crashes with these instructions.
// case HloOpcode::kDivide:
// return xnn_binary_divide;
// case HloOpcode::kMaximum:
// return xnn_binary_maximum;
case HloOpcode::kDivide:
return xnn_binary_divide;
case HloOpcode::kMaximum:
return xnn_binary_maximum;
case HloOpcode::kMinimum:
return xnn_binary_minimum;
case HloOpcode::kMultiply: