- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for UnsortedSegmentMax (0.13 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
%num_segments = "tf.Const"() {value = dense<4> : tensor<i32>} : () -> tensor<i32> // CHECK: mhlo.constant dense<-3.40282347E+38> : tensor<f32> // CHECK: mhlo.scatter // CHECK: mhlo.maximum %0 = "tf.UnsortedSegmentMax"(%data, %segment_ids, %num_segments) : (tensor<8x?x64xf32>, tensor<?x16xi32>, tensor<i32>) -> (tensor<4x?xf32>) func.return %0: tensor<4x?xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)