- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 157 for op_names (0.16 sec)
-
tensorflow/compiler/mlir/tfr/python/tfr_gen.py
def _visit_tfr_builtins(self, op_name, args, node): arg_strs = [] arg_tys = [] for arg in args: value, ty = self.visit(arg) arg_strs.append(value) arg_tys.append(ty) tfr_op_name = 'tfr.' + op_name[5:] ret_tys = ( TFR_BUILTINS[op_name](*arg_tys) if callable(TFR_BUILTINS[op_name]) else TFR_BUILTINS[op_name])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 27 15:27:03 UTC 2022 - 55.8K 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/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/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/c/experimental/ops/gen/README.md
# TensorFlow Op CodeGen Machinery (Experimental) ## Usage ``` usage: generate_cpp [flags] OpName1 [OpName2 ...] Flags: --help=false bool Print this help message. --category="" string Category for generated ops (e.g. 'math', 'array'). --namespace="" string Compact C++ namespace, default is 'tensorflow::ops'.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 21 18:51:25 UTC 2021 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/quantization_info.proto
} // The quantization parameters for a named tensor. message QuantParams { // The tensor name has the following convention: // tensor_name := op_name | op_name ’:’ port_number. // If the op has only one port, op_name can be used. // If the op has internal states, such as fused LSTM, the port_number should // follow a predefined convention. oneof name_oneof { string name = 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 08 03:45:04 UTC 2019 - 2.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 (2) -
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/stablehlo/transforms/op_stat_pass.cc
dialect_name_of_[op_with_dialect_name] = dialect_name; op_name_of_[op_with_dialect_name] = op_name; ++total_ops_; }); PrintSummary(); } void PrintOpStatsPass::CountOp( DenseMap<StringRef, llvm::StringMap<int64_t>> &op_count_map, StringRef op_name, StringRef dtype) { auto &op_counts = op_count_map[op_name]; if (auto it = op_counts.find(dtype); it != op_counts.end()) { it->second++; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc
parser.parseAttribute(device, "device", result.attributes) || parser.parseRParen())) return mlir::failure(); if (options.has_op_name && parser.parseAttribute(op_name, "op_name", result.attributes)) return mlir::failure(); if (options.has_symbol_ref && parser.parseAttribute(f, "f", result.attributes)) return mlir::failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0)