- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for registry_ (0.21 sec)
-
tensorflow/c/experimental/gradients/nn_grad_test.cc
} AbstractContextPtr immediate_execution_ctx_; GradientRegistry registry_; Status status_; 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();
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/experimental/gradients/math_grad_test.cc
} // 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_), immediate_execution_ctx_.get(), {x.get(), y.get()}, UseFunction())); }
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/gradients/array_grad_test.cc
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); status_ = RunModel(IdentityNGradModel, immediate_execution_ctx_.get(),
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
Status GradientRegistry::Register( const string& op_name, GradientFunctionFactory gradient_function_factory) { auto iter = registry_.find(op_name); if (iter != registry_.end()) { const string error_msg = "Gradient already exists for op: " + op_name + "."; return errors::AlreadyExists(error_msg); } registry_.insert({op_name, gradient_function_factory}); return absl::OkStatus(); } Status GradientRegistry::Lookup(
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/eager/gradients.h
Status Lookup(const ForwardOperation& op, std::unique_ptr<GradientFunction>* gradient_function) const; private: absl::flat_hash_map<string, GradientFunctionFactory> registry_; }; // TODO(srbs): Figure out if we can avoid declaring this in the public header. // Wrapper for a tensor output of an operation executing under a tape. //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
TFE_MonitoringCounterCellIncrementBy(cell, 1); EXPECT_EQ(TFE_MonitoringCounterCellValue(cell), 1); auto* collection_registry = monitoring::CollectionRegistry::Default(); monitoring::CollectionRegistry::CollectMetricsOptions options; std::unique_ptr<monitoring::CollectedMetrics> metrics = collection_registry->CollectMetrics(options); EXPECT_EQ("test/counter", metrics->point_set_map.at("test/counter")->metric_name);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/c_api_internal.h
}; struct TF_OperationDescription { TF_OperationDescription(TF_Graph* g, const char* op_type, const char* node_name) : node_builder(node_name, op_type, g->graph.op_registry()), graph(g) {} tensorflow::NodeBuilder node_builder; TF_Graph* graph; std::set<tensorflow::string> colocation_constraints; }; struct TF_Operation { tensorflow::Node node; private:
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sat May 13 00:49:12 GMT 2023 - 7.6K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/modular_filesystem_test.cc
// // As we need these URI schemes to instantiate the test suite when // `testing::InitGoogleTest` gets called, here we just store them to an // internal scheme registry. See `URISchemeRegister` above. static bool GetURIScheme(const std::string& scheme) { tensorflow::SchemeVector()->push_back(scheme); return true; } // This function is used for cloud filesystem
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 27 20:25:58 GMT 2022 - 71K bytes - Viewed (0) -
tensorflow/BUILD
"//tensorflow/cc/saved_model:loader_lite_impl", "//tensorflow/cc/saved_model:metrics_impl", "//tensorflow/compiler/tf2tensorrt:op_converter_registry_impl", "//tensorflow/core/common_runtime:core_cpu_impl", "//tensorflow/core/common_runtime/gpu:gpu_runtime_impl",
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 09 18:15:11 GMT 2024 - 53.4K bytes - Viewed (8) -
tensorflow/c/experimental/gradients/grad_test_helper.h
void CheckTensorValue(AbstractTensorHandle* t, absl::Span<const float> manuals, absl::Span<const int64_t> dims, double abs_error = 1e-2); Model BuildGradModel(Model forward, GradientRegistry registry); } // namespace internal } // namespace gradients } // namespace tensorflow
C - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Jan 14 20:36:51 GMT 2021 - 1.5K bytes - Viewed (0)