- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for ForwardOperation (0.24 sec)
-
tensorflow/c/eager/gradients.cc
} // Helper functions which delegate to `AbstractOperation`, update // the state of the ForwardOperation and call the tape as appropriate. // These APIs are mainly to facilitate testing and are subject to change. namespace internal { Status Reset(AbstractOperation* op_, const char* op, const char* raw_device_name, ForwardOperation* forward_op_) { forward_op_->op_name = op; forward_op_->attrs.Reset(op);
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/experimental/gradients/nn_grad.h
#include "tensorflow/c/eager/gradients.h" namespace tensorflow { namespace gradients { GradientFunction* ReluRegisterer(const ForwardOperation& op); GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer( const ForwardOperation& op); GradientFunction* BiasAddRegisterer(const ForwardOperation& op); } // namespace gradients } // namespace tensorflow
C - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Dec 03 22:28:48 GMT 2020 - 1.2K bytes - Viewed (0) -
tensorflow/c/eager/gradients_internal.h
// the state of the ForwardOperation and call the tape as appropriate. // These APIs are mainly to facilitate testing and are subject to change. // Records the op name in the `ForwardOperation`. Status Reset(AbstractOperation*, const char* op, const char* raw_device_name, ForwardOperation*); // Records the inputs in the `ForwardOperation`.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:35 GMT 2021 - 4.2K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad.h
namespace gradients { GradientFunction* AddRegisterer(const ForwardOperation& op); GradientFunction* ExpRegisterer(const ForwardOperation& op); GradientFunction* MatMulRegisterer(const ForwardOperation& op); GradientFunction* SqrtRegisterer(const ForwardOperation& op); GradientFunction* NegRegisterer(const ForwardOperation& op); GradientFunction* SubRegisterer(const ForwardOperation& op); GradientFunction* MulRegisterer(const ForwardOperation& op);
C - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Dec 03 22:28:48 GMT 2020 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
} // namespace GradientFunction* ReluRegisterer(const ForwardOperation& op) { return new ReluGradientFunction(op.outputs); } GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer( const ForwardOperation& op) { return new SparseSoftmaxCrossEntropyWithLogitsGradientFunction(op.outputs); } GradientFunction* BiasAddRegisterer(const ForwardOperation& op) { return new BiasAddGradientFunction(op.attrs); }
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) -
tensorflow/c/experimental/gradients/math_grad.cc
return new SqrtGradientFunction(op.outputs[0]); } GradientFunction* NegRegisterer(const ForwardOperation& op) { return new NegGradientFunction; } GradientFunction* SubRegisterer(const ForwardOperation& op) { return new SubGradientFunction; } GradientFunction* MulRegisterer(const ForwardOperation& op) { return new MulGradientFunction(op.inputs); }
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/not_differentiable.cc
grad_inputs[i] = nullptr; } return OkStatus(); } Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op) { return registry->Register(op, [](const ForwardOperation& op) { return new NotDifferentiableGradientFunction; }); } } // namespace gradients
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 15 01:15:58 GMT 2022 - 1.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients.h
// gradient registerer to instantiate a GradientFunction. struct ForwardOperation { public: string op_name; std::vector<AbstractTensorHandle*> inputs; std::vector<AbstractTensorHandle*> outputs; std::vector<int64_t> skip_input_indices; AttrBuilder attrs; }; using GradientFunctionFactory = std::function<GradientFunction*(const ForwardOperation& op)>; // Map from op name to a `GradientFunctionFactory`.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
int* num_retvals) { TF_RETURN_IF_ERROR(parent_op_->Execute(retvals, num_retvals)); for (int i = 0; i < *num_retvals; i++) { // TODO(srbs): Manage refcount of ForwardOperation's inputs/outputs. forward_op_.outputs.push_back(retvals[i]); } // TODO(b/166669239): This is needed to support AttrBuilder::Get for string
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/array_grad.h
#define TENSORFLOW_C_EXPERIMENTAL_GRADIENTS_ARRAY_GRAD_H_ #include "tensorflow/c/eager/gradients.h" namespace tensorflow { namespace gradients { GradientFunction* IdentityNRegisterer(const ForwardOperation& op); } // namespace gradients } // namespace tensorflow
C - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Thu Dec 03 22:28:48 GMT 2020 - 1K bytes - Viewed (0)