- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for set_is_stateful (0.38 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
} auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); } // Ignore the gradient and is_stateful attribute on the function as they have // been handled above. Ignore the entry func attribute as it is an MLIR
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); } // Ignore the gradient and is_stateful attribute on the function as they have // been handled above. Ignore the entry func attribute as it is an MLIR
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
{"shape_shape:output:0", "minval:output:0", "maxval:output:0"}, {{"Tout", DataType::DT_INT32}, {"T", DataType::DT_INT32}}}}, /*ret_def=*/{{"out", "shape:output:0"}}); func.mutable_signature()->set_is_stateful(true); *flib_def.add_function() = std::move(func); TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); NodeDef call_node; call_node.set_name("fn_call"); call_node.set_op("Stateful_func");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
auto updated_fdef = std::make_unique<FunctionDef>(); TF_RETURN_IF_ERROR( GraphToFunctionDef(*g, new_func_name, updated_fdef.get())); updated_fdef->mutable_signature()->set_is_stateful(true); const FunctionDef* original_fdef = fld->Find(func_name); if (original_fdef) { for (const auto& attr : original_fdef->attr()) { (*updated_fdef->mutable_attr())[attr.first] = attr.second;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/ops/xla_ops.cc
.Attr("Tresults: list(type) >= 0") .Attr("function: func") // XLA random-number generation ops are stateful. // TODO(phawkins): create stateful and non-stateful variants of XlaLaunch. .SetIsStateful() .Doc("XLA Launch Op. For use by the XLA JIT only."); REGISTER_OP("XlaLaunchV2") .Input("args: Targs") .Output("results: Tresults") .Attr("Targs: list(type) >= 0")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 09:08:06 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/aot/aot_only_var_handle_op.cc
)doc") .Attr("container: string = ''") .Attr("shared_name: string = ''") .Attr("debug_name: string = ''") .Attr("dtype: type") .Attr("shape: shape") .Output("resource: resource") .SetIsStateful() .SetShapeFn([](shape_inference::InferenceContext* c) { c->set_output(0, c->Scalar()); DataType t; TF_RETURN_IF_ERROR(c->GetAttr("dtype", &t)); PartialTensorShape p;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 09:57:04 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/debugging/dump_tensor_op.cc
.Input("tensor_data: T") .Attr("log_dir_path: string") .Attr("file_name: string") .Attr("T: type") .Attr("enabled: bool") .Attr("func_name: string") .Attr("node_name: string") .SetIsStateful(); class DumpTensorOp : public OpKernel { public: explicit DumpTensorOp(OpKernelConstruction* ctx) : OpKernel(ctx) { string log_dir_path; string file_name; string func_name;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 03:12:17 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op.cc
REGISTER_OP("CalibrationStatisticsSaver") .Input("args: Tin") .Attr("Tin: list(type) >= 0") .Attr("ids: list(string) >= 1") .Attr("calibration_methods: list(int) >= 1") .Attr("output_file_path: string") .SetIsStateful() .Doc(R"doc( Aggregates and saves the calibration statistics data. This op collects outputs of multiples CustomAggregator ops, which includes `min`, `max` and `histogram`. Then it aggregates them according to the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 01:31:23 UTC 2024 - 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, const char* attr_spec) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0)