Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for Tf_AnyOp (0.19 sec)

  1. tensorflow/compiler/mlir/python/mlir_wrapper/ops.cc

                  mlir::Value y) -> mlir::Operation* {
                 return opb.create<mlir::TF::AddV2Op>(loc, x, y).getOperation();
               });
    
      py::class_<mlir::TF::AnyOp>(m, "Tf_AnyOp")
          .def("create",
               [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value input,
                  mlir::Value reduction_indices,
                  bool keep_dims = false) -> 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)
  2. tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.pyi

    class Tf_AddV2Op:
        def __init__(self, *args, **kwargs) -> None: ...
        def create(self, arg0: Location, arg1: Value, arg2: Value) -> Operation: ...
    
    class Tf_AnyOp:
        def __init__(self, *args, **kwargs) -> None: ...
        def create(self, arg0: Location, arg1: Value, arg2: Value, arg3: bool) -> Operation: ...
    
    class Tf_ConstOp:
        def __init__(self, *args, **kwargs) -> None: ...
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 09 17:10:09 UTC 2023
    - 6.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

    def LegalizeProd : Pat<
      (TF_ProdOp $arg0, $axes, BoolAttr:$arg2),
      (TFL_ReduceProdOp $arg0, (CreateTFCastToInt32Op $axes), $arg2)>;
    
    def LegalizeAny : Pat<
      (TF_AnyOp $input, $reduction_indices, $keep_dims),
      (TFL_ReduceAnyOp $input, (CreateTFCastToInt32Op $reduction_indices),
                       $keep_dims)>;
    
    def LegalizeAll : Pat<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        TF_BoolTensor:$reshuffle
      );
    
      let results = (outs
        Res<TF_ResourceTensor, "", [TF_DatasetSeedGeneratorAlloc]>:$handle,
        TF_VariantTensor:$deleter
      );
    }
    
    def TF_AnyOp : TF_Op<"Any", [Pure]> {
      let summary = [{
    Computes the "logical or" of elements across dimensions of a tensor.
      }];
    
      let description = [{
    Reduces `input` along the dimensions given in `axis`. Unless
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top