- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for gradients (0.19 sec)
-
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) -
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) -
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) -
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" #include "tensorflow/c/experimental/ops/math_ops.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) -
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 = [
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) -
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
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
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
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1) -
tensorflow/c/BUILD
":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", "//tensorflow/cc/saved_model:loader_lite",
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) -
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; } 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) -
tensorflow/c/c_api_function.cc
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()); status->status = g->graph.AddGradientDef(std::move(gdef)); } int TF_GraphNumFunctions(TF_Graph* g) { tensorflow::mutex_lock l(g->mu);
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)