- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ComputeGradient (0.12 sec)
-
tensorflow/c/eager/gradients.cc
std::vector<int64_t> ids(tensors.size()); for (int i = 0; i < tensors.size(); i++) { ids[i] = ToId(tensors[i]); } return ids; } Status Tape::ComputeGradient( AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> targets, absl::Span<AbstractTensorHandle* const> sources, absl::Span<AbstractTensorHandle* const> output_gradients,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
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,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0)