Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for dynamic_conv (0.24 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

        return true;
      if (op_name == "stablehlo.broadcast_in_dim" &&
          field_name == "broadcast_dimensions")
        return true;
      if ((op_name == "stablehlo.convolution" ||
           op_name == "stablehlo.dynamic_conv") &&
          (field_name == "window_strides" || field_name == "lhs_dilation" ||
           field_name == "rhs_dilation"))
        return true;
      if (op_name == "stablehlo.dynamic_broadcast_in_dim" &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

          add = rewriter.create<tensor::CastOp>(loc, op.getType(), add);
        }
        rewriter.replaceOp(op, {add});
        return success();
      }
    };
    
    // Conterts tf.Conv2D to mhlo.dynamic_conv.
    // TODO(disc): To recover static special case's performance with adding folding,
    // canonicalization func and removing ConvertConvOp.
    template <typename OpT, int num_spatial_dims, bool depthwise_conv = false>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

    }
    
    // -----
    
    // CHECK-LABEL: conv_dynamic
    func.func @conv_dynamic(%arg0: tensor<?x32x32x6xf32>, %arg1: tensor<3x3x3x16xf32>) -> tensor<?x8x7x16xf32> {
      // CHECK: "mhlo.dynamic_conv"
    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