- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for compute (0.23 sec)
-
tensorflow/c/experimental/gradients/nn_grad.cc
: forward_outputs_(f_outputs) { for (auto output : forward_outputs_) { if (output) { output->Ref(); } } } Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { AbstractTensorHandle* upstream_grad = grad_outputs[0];
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad.cc
#include "tensorflow/c/eager/abstract_context.h" namespace tensorflow { namespace gradients { namespace { class IdentityNGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { for (int i = 0; i < grad_outputs.size(); i++) {
C++ - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 1.6K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/custom_gradient_test.cc
Status s = StatusFromTF_Status(status.get()); CHECK_EQ(errors::OK, s.code()) << s.message(); } }; class PassThroughGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { CHECK_EQ(grad_outputs.size(), 1);
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/not_differentiable.cc
==============================================================================*/ #include "tensorflow/c/experimental/gradients/not_differentiable.h" namespace tensorflow { namespace gradients { Status NotDifferentiableGradientFunction::Compute( AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) { for (int i = 0; i < grad_inputs.size(); i++) { grad_inputs[i] = nullptr; }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 15 01:15:58 GMT 2022 - 1.3K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/expiring_lru_cache_test.cc
EXPECT_EQ(num_compute_calls, 5); cache2.LookupOrCompute("c", &value, compute_func, &status); TF_EXPECT_OK(status.status); EXPECT_EQ(value, 2); EXPECT_EQ(num_compute_calls, 5); cache2.LookupOrCompute("d", &value, compute_func, &status); TF_EXPECT_OK(status.status); EXPECT_EQ(value, 3); EXPECT_EQ(num_compute_calls, 5); // Re-read "a", ensure it is re-computed. cache2.LookupOrCompute("a", &value, compute_func, &status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 09 19:31:22 GMT 2020 - 7.1K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
"If the intent is to treat this op as non-differentiable consider " "using RegisterNotDifferentiable or " "NotDifferentiableGradientFunction."); } return gradient_function->Compute(ctx_, output_gradients, result); } Status TapeVSpace::BuildOnesLike(const TapeTensor& t, AbstractTensorHandle** result) const { AbstractOperationPtr op(ctx_->CreateOperation());
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/math_grad.cc
"Expected numeric or variant tensor, got dtype ", dtype); } return tensorflow::ops::Conj(ctx, input, output, name); } class AddGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { // TODO(b/161805092): Support broadcasting.
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
control_output_names_vec.push_back(string(control_output_names[i])); } } // Compute body nodes. std::vector<const Node*> body_nodes; status->status = tensorflow::ComputeBodyNodes( fn_body, fn_name, num_opers, opers, input_nodes, &body_nodes); if (TF_GetCode(status) != TF_OK) return nullptr; // Compute body nodes. std::vector<const Node*> control_output_nodes;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
ParallelDevice::Join( const std::vector<PartialTensorShape>& expected_output_shapes, TF_Status* status) const { absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> result; // Compute per-device per-output tensors std::vector<std::vector<TensorHandlePtr>> per_device_output_tensors; per_device_output_tensors.reserve(underlying_devices_.size()); int first_op_output_count = 0;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
} GCSFile::GCSFile(google::cloud::storage::Client&& gcs_client, bool compose, uint64_t block_size, size_t max_bytes, uint64_t max_staleness, uint64_t stat_cache_max_age, size_t stat_cache_max_entries) : gcs_client(gcs_client), compose(compose), block_cache_lock(), block_size(block_size) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 23 06:55:53 GMT 2023 - 46.9K bytes - Viewed (0)