Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for grads (0.13 sec)

  1. tensorflow/c/eager/tape.h

          }
          Gradient* in_grad = in_grads[grad_index];
          if (in_grad != nullptr) {
            // ComputeGradient steals a reference
            vspace_.MarkAsResult(in_grad);
          }
          used_in_grads.push_back(in_grad);
        }
      }
    
      return tape->ComputeGradient(vspace_, targets, sources,
                                   sources_that_are_targets, used_in_grads,
                                   out_grads);
    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)
  2. tensorflow/c/c_test_util.cc

        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;
    }
    
    std::vector<string> GetFuncNames(const tensorflow::GraphDef& graph_def) {
      std::vector<string> names;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  3. tensorflow/c/eager/gradients.cc

        }
      }
    
      TF_RETURN_IF_ERROR(GradientTape::ComputeGradient(
          vspace, target_tensor_ids, source_tensor_ids, sources_that_are_targets,
          output_gradients, result, /*build_default_zeros_grads*/ false));
      return absl::OkStatus();
    }
    
    // Helper functions which delegate to `AbstractOperation`, update
    // the state of the ForwardOperation and call the tape as appropriate.
    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)
  4. tensorflow/c/experimental/gradients/math_grad.cc

                     absl::Span<AbstractTensorHandle* const> grad_outputs,
                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
        // TODO(b/161805092): Support broadcasting.
    
        DCHECK(grad_outputs[0]);
        grad_inputs[0] = grad_outputs[0];
        grad_inputs[1] = grad_outputs[0];
    
        grad_inputs[0]->Ref();
        grad_inputs[1]->Ref();
        return absl::OkStatus();
      }
    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)
  5. tensorflow/c/BUILD

                ":env",
                ":logging",
                ":tf_status",
                ":tf_tensor",
                "//tensorflow/c/experimental/filesystem:modular_filesystem",
                "//tensorflow/cc:grad_ops",
                "//tensorflow/cc:gradients",
                "//tensorflow/cc:ops",
                "//tensorflow/cc:scope_internal",
                "//tensorflow/cc:while_loop",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 27 18:00:18 GMT 2024
    - 30.3K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/gradients/math_grad_test.cc

    ==============================================================================*/
    #include "tensorflow/c/experimental/gradients/math_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"
    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)
  7. tensorflow/c/eager/BUILD

            "//tensorflow/c:c_api",
            "//tensorflow/c:c_test_util",
            "//tensorflow/c:tf_status_helper",
            "//tensorflow/c/experimental/gradients:array_grad",
            "//tensorflow/c/experimental/gradients:math_grad",
            "//tensorflow/c/experimental/gradients:not_differentiable",
            "//tensorflow/c/experimental/gradients/tape:tape_context",
            "//tensorflow/c/experimental/ops",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 33.3K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_function.cc

      if (TF_GetCode(status) != TF_OK) return;
      if (!grad) return;
    
      status->status = g->graph.AddFunctionDef(grad->record->fdef(),
                                               grad->record->stack_traces());
      if (TF_GetCode(status) != TF_OK) return;
    
      tensorflow::GradientDef gdef;
      gdef.set_function_name(func->record->fdef().signature().name());
      gdef.set_gradient_func(grad->record->fdef().signature().name());
    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)
Back to top