Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 21 for gradients (0.24 sec)

  1. tensorflow/c/eager/gradients.cc

    Status TapeVSpace::CallBackwardFunction(
        const string& op_type, GradientFunction* gradient_function,
        const std::vector<int64_t>& unneeded_gradients,
        gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
        absl::Span<AbstractTensorHandle*> result) const {
      if (gradient_function == nullptr) {
        return errors::InvalidArgument(
            "Provided null gradient_function for '", op_type, "'.\n",
    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)
  2. tensorflow/c/eager/gradients_test.cc

    #include "tensorflow/c/eager/gradients_internal.h"
    #include "tensorflow/c/eager/unified_api_testutil.h"
    #include "tensorflow/c/experimental/gradients/array_grad.h"
    #include "tensorflow/c/experimental/gradients/math_grad.h"
    #include "tensorflow/c/experimental/gradients/not_differentiable.h"
    #include "tensorflow/c/experimental/gradients/tape/tape_context.h"
    #include "tensorflow/c/experimental/ops/array_ops.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/nn_grad_test.cc

    ==============================================================================*/
    #include "tensorflow/c/experimental/gradients/nn_grad.h"
    
    #include "tensorflow/c/eager/c_api_test_util.h"
    #include "tensorflow/c/eager/unified_api_testutil.h"
    #include "tensorflow/c/experimental/gradients/grad_test_helper.h"
    #include "tensorflow/c/experimental/gradients/tape/tape_context.h"
    #include "tensorflow/c/experimental/ops/nn_ops.h"
    #include "tensorflow/c/tf_status_helper.h"
    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)
  4. tensorflow/c/experimental/gradients/array_grad_test.cc

    #include "tensorflow/c/experimental/gradients/array_grad.h"
    
    #include "tensorflow/c/eager/c_api_test_util.h"
    #include "tensorflow/c/eager/c_api_unified_experimental_internal.h"
    #include "tensorflow/c/eager/unified_api_testutil.h"
    #include "tensorflow/c/experimental/gradients/grad_test_helper.h"
    #include "tensorflow/c/experimental/gradients/tape/tape_context.h"
    #include "tensorflow/c/experimental/ops/array_ops.h"
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/tape.h

              grad_it->second.push_back(new_gradients);
              vspace.MarkAsResult(new_gradients);
            }
            out_gradients.push_back(new_gradients);
          }
        }
        VLOG(1) << "Calling gradient function for '" << trace.op_type << "'";
        std::vector<Gradient*> in_gradients(trace.input_tensor_id.size());
        DCHECK(build_default_zeros_grads || zero_indices.empty());
        if (any_gradient_nonzero) {
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  6. tensorflow/c/experimental/gradients/custom_gradient_test.cc

      std::unique_ptr<GradientFunction> gradient_function(
          new PassThroughGradientFunction);
      tape.RecordOperation(inputs, {exp_output}, gradient_function.release());
      TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx,
                                              /*targets*/ {exp_output},
                                              /*sources=*/inputs,
                                              /*output_gradients=*/{},
    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)
  7. tensorflow/c/experimental/gradients/BUILD

        hdrs = ["grad_test_helper.h"],
        visibility = ["//visibility:private"],
        deps = [
            "//tensorflow/c/eager:gradient_checker",
            "//tensorflow/c/eager:gradients_internal",
            "//tensorflow/c/eager:unified_api_testutil",
            "//tensorflow/c/experimental/gradients/tape:tape_context",
            "//tensorflow/core:test",
            "//tensorflow/core:test_main",
        ],
    )
    
    tf_cuda_cc_test(
    Plain Text
    - Registered: Tue Apr 09 12:39:09 GMT 2024
    - Last Modified: Mon Apr 01 20:39:44 GMT 2024
    - 6.7K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/gradients/grad_test_helper.cc

    ==============================================================================*/
    #include "tensorflow/c/experimental/gradients/grad_test_helper.h"
    
    #include "tensorflow/c/eager/gradient_checker.h"
    #include "tensorflow/c/experimental/gradients/tape/tape_context.h"
    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace gradients {
    namespace internal {
    
    void CompareNumericalAndAutodiffGradients(
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/gradients/nn_grad.cc

        /* Given upstream grad U and a BiasAdd: A + bias, the gradients are:
         *
         *    dA = U
         *    dbias = reduceSum(U, dims = channel_dim)
         */
    
        AbstractTensorHandle* upstream_grad = grad_outputs[0];
        DCHECK(upstream_grad);
    
        // Recover data format from forward pass for gradient.
        std::string data_format;
    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)
  10. tensorflow/c/experimental/gradients/math_grad.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/experimental/gradients/math_grad.h"
    
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/gradients.h"
    #include "tensorflow/c/experimental/ops/array_ops.h"
    #include "tensorflow/c/experimental/ops/math_ops.h"
    #include "tensorflow/c/experimental/ops/nn_ops.h"
    
    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)
Back to top