Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for compute (0.23 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top