Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for gradients (0.16 sec)

  1. tensorflow/c/eager/gradients.cc

      // Looks up the ID of a Gradient.
      int64_t TensorId(AbstractTensorHandle* tensor) const override;
    
      // Converts a Gradient to a TapeTensor.
      TapeTensor TapeTensorFromGradient(AbstractTensorHandle* g) const override;
    
      void MarkAsResult(AbstractTensorHandle* gradient) const override;
    
      void DeleteGradient(AbstractTensorHandle* gradient) const override;
    
     private:
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/BUILD

            ":c_api_unified_internal",
            "//tensorflow/c/experimental/gradients/tape:tape_operation",
            "//tensorflow/core/lib/llvm_rtti",
            "//tensorflow/core/platform:errors",
        ],
    )
    
    cc_library(
        name = "gradients_internal",
        srcs = [
            "gradients.cc",
        ],
        hdrs = [
            "gradients.h",
            "gradients_internal.h",
        ],
        visibility = [
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 11 23:52:39 UTC 2024
    - 33.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device.cc

            // (where the value starts on the host), without allowing other implicit
            // copies/broadcasts. Other implicit copies may be supported eventually,
            // but need special handling for gradients (gradient of copy-on is not
            // just copy-off but includes a sum) and consideration of performance.
            //
            // TODO(allenl): There may be smarter ways to do this copy in some
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.h

                                                            TF_Status* status);
    
    // Some TF ops need a step container to be set to limit the lifetime of some
    // resources (mostly TensorArray and Stack, used in while loop gradients in
    // graph mode). Calling this on a context tells it to start a step.
    TF_CAPI_EXPORT extern void TFE_ContextStartStep(TFE_Context* ctx);
    
    // Ends a step. When there is no active step (that is, every started step has
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  5. tensorflow/c/c_test_util.cc

    }
    
    std::vector<std::pair<string, string>> GetGradDefs(
        const tensorflow::GraphDef& graph_def) {
      std::vector<std::pair<string, string>> grads;
      for (const tensorflow::GradientDef& grad : graph_def.library().gradient()) {
        grads.emplace_back(grad.function_name(), grad.gradient_func());
      }
      std::sort(grads.begin(), grads.end());
      return grads;
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
Back to top