Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 49 for op_names (0.17 sec)

  1. 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)
  2. tensorflow/cc/framework/fuzzing/op_fuzzing.bzl

                   " $$(dirname $(location " + out_fuzz_files[0] + "))" +
                   " " + api_def_src_locations + " " + (",".join(op_names))),
        )
    
        for op_name in op_names:
            cc_test(
                name = op_name.lower() + "_fuzz",
                srcs = [op_name + "_fuzz.cc"],
                deps = kernel_deps +
                       [
                           "//tensorflow/security/fuzzing/cc:fuzz_session",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Nov 07 19:14:57 UTC 2022
    - 4.3K bytes
    - Viewed (0)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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 &region : 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)
  8. 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)
  9. 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)
  10. 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)
Back to top