Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for Register (0.22 sec)

  1. tensorflow/c/experimental/gradients/not_differentiable.cc

      for (int i = 0; i < grad_inputs.size(); i++) {
        grad_inputs[i] = nullptr;
      }
      return OkStatus();
    }
    
    Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op) {
      return registry->Register(op, [](const ForwardOperation& op) {
        return new NotDifferentiableGradientFunction;
      });
    }
    }  // namespace gradients
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Jun 15 01:15:58 GMT 2022
    - 1.3K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem_static.cc

    #include "tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.h"
    #include "tensorflow/core/platform/status.h"
    
    namespace tensorflow {
    
    // Register the POSIX filesystems statically.
    // Return value will be unused
    bool StaticallyRegisterLocalFilesystems() {
      TF_FilesystemPluginInfo info;
      TF_InitPlugin(&info);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 1.5K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/math_grad_test.cc

        ASSERT_EQ(errors::OK, status_.code()) << status_.message();
        y.reset(y_raw);
      }
    
      // TODO(srbs): Rename ops::Add to ops::AddV2 and AddRegister to
      // AddV2Registerer.
      status_ = registry_.Register("AddV2", AddRegisterer);
      ASSERT_EQ(errors::OK, status_.code()) << status_.message();
    
      ASSERT_NO_FATAL_FAILURE(CompareNumericalAndAutodiffGradients(
          AddModel, BuildGradModel(AddModel, registry_),
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Thu Apr 13 17:32:14 GMT 2023
    - 16.3K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/gradients/array_grad_test.cc

            immediate_execution_ctx_.get(), 1.0f, &x2_raw);
        ASSERT_EQ(errors::OK, status_.code()) << status_.message();
        x2.reset(x2_raw);
      }
    
      status_ = registry_.Register("IdentityN", IdentityNRegisterer);
      ASSERT_EQ(errors::OK, status_.code()) << status_.message();
      auto IdentityNGradModel = BuildGradModel(IdentityNModel, registry_);
    
      std::vector<AbstractTensorHandle*> outputs(2);
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/filesystem/modular_filesystem_registration.cc

      if (ops == nullptr)
        return errors::FailedPrecondition(
            "Trying to register filesystem without operations");
    
      if (ops->init == nullptr)
        return errors::FailedPrecondition(
            "Trying to register filesystem without `init` operation");
    
      if (ops->cleanup == nullptr)
        return errors::FailedPrecondition(
            "Trying to register filesystem without `cleanup` operation");
    
      return OkStatus();
    }
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 07 22:08:43 GMT 2023
    - 12.8K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_distributed_test.cc

    // be thrown in the registered graph optimization pass.
    TEST(CAPI, DistributedFunctionGraphPassOnlyOnce) {
      // Register graph pass that will raise error if called more than once.
      tensorflow::optimization_registration::OptimizationPassRegistration
          register_test_pass(tensorflow::OptimizationPassRegistry::PRE_PLACEMENT, 0,
                             std::make_unique<GraphErrorInjectionPass>(),
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  7. tensorflow/c/eager/gradients.cc

      TF_RETURN_IF_ERROR(
          op->Execute(absl::Span<AbstractTensorHandle*>(outputs), &num_outputs));
      *result = outputs[0];
      return absl::OkStatus();
    }
    }  // namespace
    
    Status GradientRegistry::Register(
        const string& op_name, GradientFunctionFactory gradient_function_factory) {
      auto iter = registry_.find(op_name);
      if (iter != registry_.end()) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/gradients/nn_grad_test.cc

     public:
      bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; }
      bool UseFunction() const { return std::get<2>(GetParam()); }
    };
    
    TEST_P(CppGradients, TestReluGrad) {
      status_ = registry_.Register("Relu", ReluRegisterer);
      ASSERT_EQ(errors::OK, status_.code()) << status_.message();
    
      auto ReluGradModel = BuildGradModel(ReluModel, registry_);
    
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 8.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

    // `first_device` and `second_device`.
    void BasicTestsForTwoDevices(TFE_Context* context, const char* first_device,
                                 const char* second_device) {
      // Register the custom device
      std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
          TF_NewStatus(), TF_DeleteStatus);
      const char* device_name = "/job:localhost/replica:0/task:0/device:CUSTOM:0";
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      string name_;
    };
    
    static TracingContext* GraphTracingFactory(const char* name, TF_Status* s) {
      return new GraphContext(name);
    }
    
    // Register the tracing implemented in this file as the default tracing engine.
    static bool register_tracing = [] {
      RegisterTracingEngineFactory("graphdef", GraphTracingFactory);
      SetDefaultTracingEngine("graphdef").IgnoreError();
      return true;
    }();
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
Back to top