- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for inp (0.02 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// dimensions with max as the reduction function. // // Sample result for VALID padding mode: // // %init = arith.constant dense<...> : tensor<i32> // %max_pool = "mhlo.reduce"(%inp, %init) ["mhlo.maximum"] // {window_dimensions = ..., window_strides = ... } // template <typename OpTy, int num_dims> class ConvertMaxPoolOp : public OpRewritePattern<OpTy> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-NEXT: %[[LEAKY:.*]] = mhlo.multiply %[[INP:.*]], %[[ALPHA]] : tensor<1x4x4x3xf32> // CHECK-NEXT: %[[CMP:.*]] = mhlo.compare GT, %[[INP]], %[[ZERO]], NOTYPE : (tensor<1x4x4x3xf32>, tensor<1x4x4x3xf32>) -> tensor<1x4x4x3xi1> // CHECK-NEXT: %[[RES:.*]] = mhlo.select %[[CMP]], %[[INP]], %[[LEAKY]] : tensor<1x4x4x3xi1>, tensor<1x4x4x3xf32> // CHECK-NEXT: return %[[RES]] : tensor<1x4x4x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)