Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for inp (0.06 sec)

  1. 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)
  2. 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)
Back to top