Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for ResizeBilinear (0.27 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/tests/odml-to-stablehlo-smuggle-resize.mlir

          // CHECK: %{{.*}} = stablehlo.custom_call @tf.ResizeBilinear(%arg0, %{{.*}}) {align_corners = false, device = "", half_pixel_centers = true} : (tensor<1x32x32x128xf32>, tensor<2xi32>) -> tensor<1x64x64x128xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 18:33:43 UTC 2024
    - 1.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo_custom_call.mlir

      %0 = "vhlo.custom_call_v1" (%arg0, %arg1) <{api_version = #vhlo<api_version_v1 API_VERSION_ORIGINAL>, 
        backend_config = #vhlo.string_v1<"">,
        call_target_name = #vhlo.string_v1<"tf.ResizeBilinear">,
        called_computations = #vhlo.array_v1<[]>,
        has_side_effect = #vhlo.bool_v1<false>,
        operand_layouts = #vhlo.array_v1<[]>,
        output_operand_aliases = #vhlo.array_v1<[]>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 14 19:15:40 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/transforms.h

    namespace odml {
    
    // Adds all the necessary passes to lower a TF module to StableHLO.
    // `skip_resize` enables or disables skipping conversion of tf.ResizeBilinear
    // and tf.ResizeNearestNeighbor ops.
    // `smuggle_disallowed_ops` enables or disables converting disallowed ops
    // like tf.ResizeBilinear or tf.ResizeNearestNeighbor to mhlo.custom_call ops.
    void AddTFToStablehloPasses(OpPassManager& pm, bool skip_resize,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 01:08:27 UTC 2024
    - 1.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/tests/tfl_legalize_hlo_custom_call.mlir

    func.func @mhlo_custom_call_does_not_legalize_tf_function(%arg0: tensor<1x4xf32>) -> tensor<1x8xf32> {
      %0 = mhlo.custom_call @tf.ResizeBilinear(%arg0) {
        backend_config = "this_is_a_test_string"
      } : (tensor<1x4xf32>) -> (tensor<1x8xf32>)
      func.return %0 : tensor<1x8xf32>
    
      //       CHECK: %0 = mhlo.custom_call @tf.ResizeBilinear(%arg0) {
      //  CHECK-SAME:   backend_config = "this_is_a_test_string"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 2.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc

    namespace mlir {
    namespace odml {
    
    namespace {
    
    // Convert op to stablehlo.custom_call
    //   "tf.ResizeBilinear"(%341, %138) {
    //      align_corners = false, device = "", half_pixel_centers = true}
    //   ==>
    //   stablehlo.custom_call @tf.ResizeBilinear(%arg0, %arg1) {
    //      align_corners = false, device = "", half_pixel_centers = true}
    LogicalResult SmuggleOp(Operation* op, PatternRewriter& rewriter) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 19 01:14:04 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc

      Option<bool> skip_quantization_ops_{
          *this, "skip-quantization-ops",
          ::llvm::cl::desc("Skip quantization ops")};
    
      Option<bool> skip_resize_{
          *this, "skip-resize",
          ::llvm::cl::desc("Skip tf.ResizeBilinear and tf.ResizeNearestNeighbor")};
    
      Option<bool> skip_partitioned_calls_{
          *this, "skip-partitioned-calls",
          ::llvm::cl::desc(
              "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")};
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 7.5K bytes
    - Viewed (0)
Back to top