- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 46 for op_names (0.2 sec)
-
tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen_main.cc
std::vector<string> op_names) { OpList ops; absl::StatusOr<ApiDefMap> api_def_map = LoadOpsAndApiDefs(ops, false, api_def_dirs); TF_CHECK_OK(api_def_map.status()); Env* env = Env::Default(); absl::Status status; std::unique_ptr<WritableFile> fuzz_file = nullptr; for (const OpDef& op_def : ops.op()) { if (std::find(op_names.begin(), op_names.end(), op_def.name()) == op_names.end())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 10:53:49 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/op_fuzzing.bzl
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 07 19:14:57 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/common/path_config.cc
namespace generator { PathConfig::PathConfig(const string& output_dir, const string& source_dir, const string& api_dir_list, const std::vector<string> op_names) : output_path(output_dir), op_names(op_names) { api_dirs = str_util::Split(api_dir_list, ",", str_util::SkipEmpty()); // Decompose the directory components given the output/source directories. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 09:51:28 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/common/controller.cc
void Controller::BuildModel() { // Build the internal data model for the requested ops for (const auto& op_name : path_config_.op_names) { const OpDef* op_def = nullptr; TF_CHECK_OK(OpRegistry::Global()->LookUpOpDef(op_name, &op_def)); CHECK(op_def != nullptr); // Crash OK const ApiDef* api_def = api_def_map_->GetApiDef(op_name); CHECK(api_def != nullptr); // Crash OK
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 09:51:28 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/cc/framework/scope_internal.h
bool copy_names); Impl(const Scope& other, Tags::OpName, const string& name, const string& op_name); Impl(const Scope& other, Tags::ControlDeps, std::vector<Operation> control_deps, bool clear_control_deps); Impl(const Scope& other, Tags::Device, const string& device); Impl(const Scope& other, Tags::SingleUseScope, const string& op_name); Impl(const Scope& other, Tags::ExitOnError);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 15:46:43 UTC 2022 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst.cc
// Find the op names with tf or tfl dialect prefix, Ex: "tf.Abs" or "tfl.Abs". auto collectOps = [this](Operation *op) { const auto &op_name = op->getName().getStringRef().str(); if (absl::StartsWith(op_name, "tf.") || absl::StartsWith(op_name, "tfl.")) { loc_to_name_.emplace(op->getLoc(), op_name); } }; for (auto ®ion : module->getRegions()) { region.walk(collectOps); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 5.3K bytes - Viewed (0) -
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/lite/stablehlo/transforms/check_accepted_ops_pass.cc
getOperation()->walk([&](Operation *op) { auto dialect_name = op->getDialect()->getNamespace(); auto op_name = op->getName().stripDialect(); if (IsAcceptedOp(dialect_name, op_name, accepted_dialects_)) { // If given op is in the `accepted_dialects_`, it's ok. } else if (IsAcceptedOp(dialect_name, op_name, optional_accepted_dialects_)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 21:06:11 UTC 2023 - 2.8K bytes - Viewed (0)