- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 49 for op_names (0.17 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/tfr/python/composite.py
# TODO(fengliuai): support input_binding and output_binding so the arguments # are not positional. def __init__(self, op_name, inputs=None, attrs=None, derived_attrs=None, outputs=None): self._op_name = op_name self._inputs = inputs self._attrs = attrs self._derived_attrs = derived_attrs self._outputs = outputs
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 1.9K 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/compiler/mlir/tfr/python/op_reg_gen.py
all_dec_args.update(kw_dec_args) op_name = all_dec_args['op_name'] op_def = op_def_registry.get(op_name) if op_def: if len(all_dec_args) > 1: # Op has been registered, so it is a user error to specify op def. raise ValueError('Op has been registered: ' + op_name) else: # Op has been registered, then we don't need to generate register code.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 5K 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)