Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for fused_ops (0.14 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/fused_kernel_matcher.mlir

      // CHECK: %[[VAL_0:.*]] = "tf._FusedConv2D"(%arg2, %arg1, %arg0) <{data_format = "NHWC", dilations = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_ops = ["BiasAdd"], num_args = 1 : i64, operandSegmentSizes = array<i32: 1, 1, 1, 0>, padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}> {TArgs = [f32]} : (tensor<8x32x32x3xf32>, tensor<1x1x3x128xf32>, tensor<128xf32>) -> tensor<*xf32>...
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 13.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        if (bias_add_op) rewriter.setInsertionPoint(bias_add_op);
    
        Value fused_op = rewriter.create<FusedOpT>(fused_loc, result_type,
                                                   ValueRange(operands), attrs);
        auto op_to_replace = fuse_activation ? activation : bias_add;
        rewriter.replaceOp(op_to_replace, ValueRange({fused_op}));
        return success();
      }
    };
    
    const char kDeviceAttr[] = "device";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir

      %conv2d = "tf._FusedConv2D"(%input, %filter, %bias, %act, %input_scale, %side_input_scale) {
        data_format = "NHWC", dilations = [1, 1, 1, 1], epsilon = 9.99999974E-5 : f32, explicit_paddings = [], filter_format = "HWIO", fused_ops = ["BiasAdd", "Relu"], leakyrelu_alpha = 2.000000e-01 : f32, num_args = 2 : i64, operandSegmentSizes = array<i32: 1, 1, 2, 2>, padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 15.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

      toco_flags.set_reduce_type_precision(reduce_type_precision);
      // Read list of user select ops.
      llvm::SmallVector<llvm::StringRef, 2> user_ops;
      (llvm::StringRef(select_user_tf_ops))
          .split(user_ops, ',', /*MaxSplit=*/-1,
                 /*KeepEmpty=*/false);
      llvm::for_each(user_ops, [&toco_flags](llvm::StringRef op_name) {
        *(toco_flags.add_select_user_tf_ops()) = op_name.str();
      });
    
      std::string result;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 14K bytes
    - Viewed (0)
  5. tensorflow/cc/BUILD

            "manip_ops",
            "nn_ops",
            "no_op",
            "parsing_ops",
            "random_ops",
            "sparse_ops",
            "state_ops",
            "string_ops",
            "training_ops",
            "user_ops",
        ],
        other_hdrs = [
            "ops/array_ops.h",
            "ops/const_op.h",
            "ops/math_ops.h",
            "ops/dataset_ops.h",
            "ops/experimental_dataset_ops.h",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 27 18:00:18 UTC 2024
    - 23.5K bytes
    - Viewed (0)
Back to top