- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for Register (0.2 sec)
-
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) -
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/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/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 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 07 22:08:43 GMT 2023 - 12.8K 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/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/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/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/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)