- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for BuildGradModel (0.3 sec)
-
tensorflow/c/experimental/gradients/math_grad_test.cc
Log1pModel, BuildGradModel(Log1pModel, registry_), immediate_execution_ctx_.get(), {x.get()}, UseFunction())); } TEST_P(CppGradients, TestDivNoNanGrad) { status_ = registry_.Register("DivNoNan", DivNoNanRegisterer); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); auto DivNoNanGradModel = BuildGradModel(DivNoNanModel, registry_); AbstractTensorHandlePtr x;
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/nn_grad_test.cc
}; TEST_P(CppGradients, TestReluGrad) { status_ = registry_.Register("Relu", ReluRegisterer); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); auto ReluGradModel = BuildGradModel(ReluModel, registry_); float X_vals[] = {1.0f, 2.0f, 3.0f, -5.0f, -4.0f, -3.0f, 2.0f, 10.0f, -1.0f}; int64_t X_dims[] = {3, 3}; AbstractTensorHandlePtr X; { AbstractTensorHandle* X_raw;
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/grad_test_helper.h
double abs_error = 1e-2); 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) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
} else { ASSERT_NEAR(manuals[j], danalytical[j], abs_error); } } TF_DeleteTensor(analytical_tensor); delete[] danalytical; } Model BuildGradModel(Model forward, GradientRegistry registry) { return [forward_model = std::move(forward), grad_registry = std::move(registry)]( AbstractContext* ctx,
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/experimental/gradients/array_grad_test.cc
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)