Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for Tf_LegacyCallOp (0.14 sec)

  1. tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.pyi

    class Tf_GreaterOp:
        def __init__(self, *args, **kwargs) -> None: ...
        def create(self, arg0: Location, arg1: Value, arg2: Value) -> Operation: ...
    
    class Tf_LegacyCallOp:
        def __init__(self, *args, **kwargs) -> None: ...
        def create(self, arg0: Location, arg1: list[Type], arg2: list[Value], arg3: str) -> Operation: ...
    
    class Tf_LessEqualOp:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 09 17:10:09 UTC 2023
    - 6.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/python/mlir_wrapper/ops.cc

                  mlir::Value y) -> mlir::Operation* {
                 return opb.create<mlir::TF::GreaterOp>(loc, x, y).getOperation();
               });
    
      // mlir::TF::LegacyCallOp
      py::class_<mlir::TF::LegacyCallOp>(m, "Tf_LegacyCallOp")
          .def("create",
               [](mlir::OpBuilder& opb, mlir::Location loc,
                  std::vector<mlir::Type> output, std::vector<mlir::Value> args,
                  std::string f) -> mlir::Operation* {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 30 02:12:49 UTC 2022
    - 7.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td

                (CallOp $f, $args),
              [(ArgTypesMatchCallee $op, $args, $f)]>;
    }
    
    // The extra attr on this op is _disable_call_shape_inference, which we ignore
    // in the bridge.
    def : Pat<(TF_LegacyCallOp:$op $args, FlatSymbolRefAttr:$f, $attr),
              (CallOp $f, $args),
            [(ArgTypesMatchCallee $op, $args, $f)]>;
    
    //===----------------------------------------------------------------------===//
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 34.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

      TF_DerivedOperandTypeListAttr Tnext_func_args = TF_DerivedOperandTypeListAttr<1>;
      TF_DerivedOperandTypeListAttr Tfinalize_func_args = TF_DerivedOperandTypeListAttr<2>;
    }
    
    def TF_LegacyCallOp : TF_Op<"LegacyCall",
          [CallOpInterface,
           DeclareOpInterfaceMethods<SymbolUserOpInterface>, Pure]> {
      let summary =
        "returns `f(inputs)`, where `f` is a function.";
    
      let description = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
Back to top