- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SetIsAggregate (0.27 sec)
-
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) -
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) -
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) -
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)