- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetStatefulAttrName (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_dialect.h
static StringRef GetGradientAttrName() { return "tf.gradient"; } // This attribute marks if a function is stateful. // Returns the string description of stateful attribute. static StringRef GetStatefulAttrName() { return "tf.signature.is_stateful"; } // Returns true if the op can be duplicated during transformations. static bool CanDuplicate(Operation *op); // Returns true if the op can have side effects.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
grad.set_function_name(function_name); grad.set_gradient_func(grad_func.getName().str()); TF_RETURN_IF_ERROR(flib_def->AddGradientDef(grad)); } auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); }
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
grad.set_function_name(function_name); grad.set_gradient_func(grad_func.getName().str()); TF_RETURN_IF_ERROR(flib_def->AddGradientDef(grad)); } auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); }
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/mlir/tensorflow/translate/import_model.cc
} // Checks opdef stateful attribute and import that as Function Attribute if (func_def->signature().is_stateful()) { auto stateful_str = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); attributes.push_back( builder_.getNamedAttr(stateful_str, builder_.getUnitAttr())); } // Checks for an associated custom gradient function. Adds it to the attribute
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)