Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 10 for Register (0.48 sec)

  1. 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)
  2. 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)
  3. 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 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  4. 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 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  5. 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)
  6. 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 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  7. tensorflow/c/c_api_test.cc

      REGISTER_OP("CApiAttributesTestOpList" #type)               \
          .Attr("v: list(" #type ")")                             \
          .SetShapeFn(tensorflow::shape_inference::UnknownShape)
    ATTR_TEST_REGISTER_OP(string);
    ATTR_TEST_REGISTER_OP(int);
    ATTR_TEST_REGISTER_OP(float);
    ATTR_TEST_REGISTER_OP(bool);
    ATTR_TEST_REGISTER_OP(type);
    ATTR_TEST_REGISTER_OP(shape);
    ATTR_TEST_REGISTER_OP(tensor);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  8. tensorflow/c/experimental/filesystem/modular_filesystem.cc

    // TODO(b/139060984): After all filesystems are converted, all calls to
    // methods from `FileSystem` will have to be replaced to calls to private
    // methods here, as part of making this class a singleton and the only way to
    // register/use filesystems.
    
    namespace tensorflow {
    
    using UniquePtrTo_TF_Status =
        ::std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)>;
    
    Status ModularFileSystem::NewRandomAccessFile(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 23.1K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_test.cc

    }
    TEST(CAPI, ExecuteWithTracing) { ExecuteWithTracing(false); }
    TEST(CAPI, ExecuteWithTracingAsync) { ExecuteWithTracing(true); }
    
    REGISTER_OP("TestNonCommUnavailable")
        .Output("out: string")
        .Doc(R"doc(Test non-communication op throwing Unavailable error.)doc");
    
    REGISTER_OP("TestCommUnavailable")
        .Output("out: string")
        .SetIsDistributedCommunication()
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  10. tensorflow/c/c_api_function_test.cc

                                 /*opts=*/nullptr, description, s.get());
      ASSERT_EQ(TF_OK, TF_GetCode(s.get())) << TF_Message(s.get());
      ASSERT_NE(*func, nullptr);
    }
    
    REGISTER_OP("CustomOp")
        .Output("output: float32")
        .Attr("index: int")
        .SetShapeFn(tensorflow::shape_inference::UnknownShape);
    
    void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
Back to top