Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for user_ops (0.28 sec)

  1. 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)
  2. 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)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc

        Operation* user_op, int argument_index, ElementsAttr value,
        llvm::SmallVector<std::pair<Region*, int>, 4>* work_list,
        llvm::MapVector<Operation*, llvm::SmallVector<unsigned int, 4>>*
            arguments_to_erase) {
      if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(user_op)) {
        (*arguments_to_erase)[read_variable_op];
        PropagateUsage(read_variable_op, value);
      } else if (auto call = dyn_cast<CallOpInterface>(user_op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 09:56:53 UTC 2024
    - 19.4K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        // together; epsilon (only with FusedBatchNorm).
        std::vector<NamedAttribute> attrs = contraction->getAttrs();
        ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops);
        attrs.push_back(
            NamedAttribute(StringAttr::get(context, "fused_ops"), fused_ops_attr));
        // Epsilon is used only in fusions with the FusedBatchNorm op, so we zero it
        // here.
        Attribute epsilon = rewriter.getF32FloatAttr(0);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  6. 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)
Back to top