- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 58 for registry_ (0.17 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/tape/tape_context.cc
namespace tensorflow { namespace gradients { TapeContext::TapeContext(AbstractContext* c, Tape* tape, const GradientRegistry& registry) : AbstractContext(kTape), parent_ctx_(c), tape_(tape), registry_(registry) { // TODO(srbs): Make AbstractContext ref counted. // parent_ctx_->Ref(); } void TapeContext::Release() { // TODO(srbs): Change to Unref() delete this; }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 1.7K 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) -
common/scripts/setup_env.sh
elif [[ $LOCAL_OS == Darwin ]]; then TARGET_OS=darwin readlink_flags="" else echo "This system's OS, $LOCAL_OS, isn't supported" exit 1 fi # Build image to use TOOLS_REGISTRY_PROVIDER=${TOOLS_REGISTRY_PROVIDER:-gcr.io} PROJECT_ID=${PROJECT_ID:-istio-testing} if [[ "${IMAGE_VERSION:-}" == "" ]]; then IMAGE_VERSION=master-b0f2fd3b4240c8178b14de4689d0e663e11868ff fi if [[ "${IMAGE_NAME:-}" == "" ]]; then
Shell Script - Registered: Wed May 08 22:53:08 GMT 2024 - Last Modified: Thu May 02 16:31:40 GMT 2024 - 7.6K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_context.h
return ptr->getKind() == kTape; } ~TapeContext() override; private: AbstractContext* parent_ctx_; // Not owned. Tape* tape_; const GradientRegistry& registry_; }; } // namespace gradients } // namespace tensorflow
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 1.6K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
namespace tensorflow { namespace gradients { TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape, const GradientRegistry& registry) : AbstractOperation(kTape), parent_op_(parent_op), tape_(tape), registry_(registry) { // TODO(b/172003047): Consider making AbstractOperation RefCounted. // parent_op_->Ref(); } void TapeOperation::Release() {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/tape/tape_operation.h
return ptr->getKind() == kTape; } ~TapeOperation() override; private: AbstractOperation* parent_op_; ForwardOperation forward_op_; Tape* tape_; const GradientRegistry& registry_; }; } // namespace gradients } // namespace tensorflow
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 3.7K bytes - Viewed (1) -
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) -
android/guava-tests/test/com/google/common/eventbus/SubscriberRegistryTest.java
registry.register(new ObjectSubscriber()); assertEquals(3, Iterators.size(registry.getSubscribers(""))); assertEquals(1, Iterators.size(registry.getSubscribers(new Object()))); assertEquals(1, Iterators.size(registry.getSubscribers(1))); registry.register(new IntegerSubscriber()); assertEquals(3, Iterators.size(registry.getSubscribers("")));
Java - Registered: Fri May 03 12:43:13 GMT 2024 - Last Modified: Wed Sep 06 17:04:31 GMT 2023 - 5.7K bytes - Viewed (0) -
maven-core/src/main/java/org/apache/maven/lifecycle/DefaultLifecycles.java
this.customLifecycles = lifecycles; this.lookup = null; this.registry = null; } @Inject public DefaultLifecycles(LifecycleRegistry registry, Lookup lookup) { this.lookup = lookup; this.registry = registry; } /** * Get lifecycle based on phase * * @param phase * @return */
Java - Registered: Sun May 05 03:35:11 GMT 2024 - Last Modified: Fri Apr 12 10:50:18 GMT 2024 - 6K bytes - Viewed (0)