- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 96 for op_names (0.16 sec)
-
tensorflow/c/eager/tracing_utils.cc
namespace tracing { Status MaybeSetOpName(AbstractOperation* op, const char* op_name) { if (isa<TracingOperation>(op)) { TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name)); } if (isa<gradients::TapeOperation>(op)) { TF_RETURN_IF_ERROR(MaybeSetOpName( dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(), op_name)); } return absl::OkStatus(); } } // namespace tracing
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils_test.cc
using tensorflow::OpRegistrationData; using tensorflow::OpRegistry; using tensorflow::Status; namespace { void Register(const std::string& op_name, OpRegistry* registry) { registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { op_reg_data->op_def.set_name(op_name); return absl::OkStatus(); }); } } // namespace TEST(TfTextUtilsTest, TestTfTextRegistered) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:14:37 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc
} } // namespace void RegisterCostFunction(absl::string_view op_name, CostFunction cost_function) { RegisterCostFunction(GetCostFunctionRegistry(), op_name, std::move(cost_function)); } bool HasCostFunctionRegistered(absl::string_view op_name) { return GetCostFunctionRegistry().contains(op_name); } int64_t CostAnalysis::GetCost(mlir::Operation* op) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc
"/tensorflow/core/tf2xla/v1/mlir_failed_xla_legalize_tf_count", "Counts the attempts of legalization of ops", "op_name"); auto *mlir_failed_legalization_count = tensorflow::monitoring::Counter<2>::New( "/tensorflow/core/tf2xla/v1/mlir_failed_xla_legalize_tf_pass_count", "Counts the failure of legalization of ops", "op_name", "legality"); class LegalizeTF : public impl::LegalizeTFBase<LegalizeTF> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
bool IsAllowListedOp(Operation *op) { std::string op_name = op->getName().getStringRef().str(); if (IsAlwaysAllowlistedOp(op) || IsTfFakeQuantOp(op)) { return true; } else if (mode_ == kDefaultMode) { return QuantizableOpsInDefaultMode().count(op_name) > 0; } else if (mode_ == kLegacyIntegerMode) { return QuantizableOpsInLegacyMode().count(op_name) > 0; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py
def _find_op( graph: ops.Graph, op_name: Optional[str] ) -> Optional[ops.Operation]: """Finds the operation with `op_name`. Args: graph: The graph to find from. op_name: Name of the node. Returns: The operation that corresponds to `op_name`. Returns None iff op_name is an empty string or None. Raises: ValueError: `op_name` is malformed. """ if not op_name:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
} } // namespace Status GradientRegistry::Register( const string& op_name, GradientFunctionFactory gradient_function_factory) { auto iter = registry_.find(op_name); if (iter != registry_.end()) { const string error_msg = "Gradient already exists for op: " + op_name + "."; return errors::AlreadyExists(error_msg); } registry_.insert({op_name, gradient_function_factory}); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/types_util.cc
const std::string& op_name, const Location& location) { using tflite::metrics::ConverterErrorData; ConverterErrorData error; if (!pass_name.empty()) { error.set_subcomponent(pass_name); } if (!error_message.empty()) { error.set_error_message(error_message); } if (!op_name.empty()) { error.mutable_operator_()->set_name(op_name); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/jit/clone_constants_for_better_clustering.cc
absl::StatusOr<bool> IsSmallConstant(Node* n) { if (!n->IsConstant()) { return false; } return IsConstantSmall(n); } bool IsInPlaceOp(absl::string_view op_name) { return op_name == "InplaceUpdate" || op_name == "InplaceAdd" || op_name == "InplaceSub"; } } // namespace Status CloneConstantsForBetterClusteringPassImpl::CloneSmallConstantInputs( const absl::flat_hash_set<string>& name_set, Node* n) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
return true; } } return false; } // Returns whether the current op is not supported by the TF Lite runtime. static bool IsUnsupportedFlexOp(const std::string& op_name) { return op_name == "PartitionedCall" || op_name == "StatefulPartitionedCall"; } // Create description of operation that could not be converted. static std::string GetOpDescriptionForDebug(Operation* inst) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)