Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for SetIsAggregate (0.27 sec)

  1. tensorflow/compiler/mlir/tfr/resources/composite_ops.cc

    namespace tensorflow {
    
    REGISTER_OP("MyAddN")
        .Input("inputs: N * T")
        .Output("sum: T")
        .Attr("N: int >= 1")
        .Attr("T: {numbertype, variant}")
        .SetIsCommutative()
        .SetIsAggregate();
    
    REGISTER_OP("MyBiasedDense")
        .Input("input: T")
        .Input("weight: T")
        .Input("bias: T")
        .Output("out: T")
        .Attr("T: {float, int8}")
        .Attr("act: {'', 'relu', 'relu6'} = ''");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 23 21:28:23 UTC 2020
    - 1.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx_test.cc

    namespace tensorflow {
    namespace {
    
    REGISTER_OP("MyAddN")
        .Input("inputs: N * T")
        .Output("sum: T")
        .Attr("N: int >= 1")
        .Attr("T: {numbertype, variant}")
        .SetIsCommutative()
        .SetIsAggregate()
        .SetShapeFn(shape_inference::UnchangedShape);
    
    REGISTER_OP("RiscAddDummy")
        .Input("x: T")
        .Input("y: T")
        .Output("z: T")
        .Attr(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 5.8K bytes
    - Viewed (0)
  3. tensorflow/c/ops.cc

        reinterpret_cast<OpDefBuilder*>(builder)->func_name();                \
      }
    
    DEFINE_BUILDER_BOOL_SETTER(SetIsCommutative)
    DEFINE_BUILDER_BOOL_SETTER(SetIsAggregate)
    DEFINE_BUILDER_BOOL_SETTER(SetIsStateful)
    DEFINE_BUILDER_BOOL_SETTER(SetAllowsUninitializedInput)
    
    void TF_OpDefinitionBuilderAddAttr(TF_OpDefinitionBuilder* builder,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 28 22:41:35 UTC 2022
    - 10.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

    REGISTER_OP("AddNLikeTest")
        .Input("inputs: N * T")
        .Output("sum: T")
        .Attr("N: int >= 1")
        .Attr("T: numbertype")
        .SetIsCommutative()
        .SetIsAggregate();
    
    Node* Sequencer(const GraphDefBuilder::Options& opts,
                    const string& call_node_name) {
      if (opts.HaveError()) return nullptr;
      NodeBuilder node_builder(opts.GetNameForOp("NoOp"), "NoOp",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
Back to top