Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for has_op_name (0.21 sec)

  1. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.h

            *op_attrs);
    
    struct ParseExecuteOpOptions {
      bool has_chain = false;
      bool has_key = false;
      bool has_device = false;
      bool has_func_attr = false;
      bool has_cost = false;
      bool has_op_name = true;
      bool has_symbol_ref = false;
    };
    
    mlir::ParseResult ParseExecuteOpCommon(mlir::OpAsmParser &parser,
                                           mlir::Builder &builder,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc

          (parser.parseKeyword("device") || parser.parseLParen() ||
           parser.parseAttribute(device, "device", result.attributes) ||
           parser.parseRParen()))
        return mlir::failure();
    
      if (options.has_op_name &&
          parser.parseAttribute(op_name, "op_name", result.attributes))
        return mlir::failure();
    
      if (options.has_symbol_ref &&
          parser.parseAttribute(f, "f", result.attributes))
        return mlir::failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc

      parse_options.has_chain = false;
      parse_options.has_key = false;
      parse_options.has_device = true;
      parse_options.has_func_attr = false;
      parse_options.has_cost = false;
      parse_options.has_op_name = false;
      parse_options.has_symbol_ref = true;
    
      auto &builder = parser.getBuilder();
      return fallback_common::ParseExecuteOpCommon(
          parser, builder, result, builder.getType<fallback::TFTensorType>(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 15.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc

      int mlir_only_patterns = 0;
      for (auto& pattern : mlir_legalize_lower_patterns.getNativePatterns()) {
        std::optional<OperationName> pat_op_name = pattern->getRootKind();
        if (!pat_op_name) {
          continue;
        }
    
        if (!HasTf2XlaFallback(pat_op_name->getRegisteredInfo()->getTypeID())) {
          mlir_only_patterns++;
        }
      }
    
      EXPECT_EQ(mlir_only_patterns, 63);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 30 03:31:01 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  5. tensorflow/cc/ops/const_op.h

      // can be directly called here instead of adding the Cast op to the graph.
      auto orig_const = AsNodeOut(scope, orig_const_output);
      const auto cast_op_name = scope.GetUniqueNameForOp("Cast");
    
      auto cast_builder = NodeBuilder(cast_op_name, "Cast")
                              .Input(orig_const)
                              .Attr("DstT", DataTypeToEnum<DstT>::v());
      scope.UpdateBuilder(&cast_builder);
      Node* ret;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 17 09:17:01 UTC 2020
    - 2.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc

      for (auto &pattern : from.getNativePatterns()) {
        std::optional<OperationName> pat_op_name = pattern->getRootKind();
        // If the pattern does not have a specific operation, always include it,
        // If the pattern is in include_ops then include it.
        bool include =
            !pat_op_name ||
            IsTypeLegalizedWithMlir(pat_op_name->getRegisteredInfo()->getTypeID());
        if (include) to.add(std::move(pattern));
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 10.2K bytes
    - Viewed (0)
Back to top