- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for Register (0.48 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)