- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for AttrBuilder (0.26 sec)
-
tensorflow/c/experimental/gradients/nn_grad.cc
private: vector<AbstractTensorHandle*> forward_outputs_; }; // TODO(vnvo2409): Add python test class BiasAddGradientFunction : public GradientFunction { public: explicit BiasAddGradientFunction(AttrBuilder f_attrs) : forward_attrs_(f_attrs) {} Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs,
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/tape/tape_operation.cc
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 // attributes. Number type attrs and DataType attrs work fine without this. // Consider getting rid of this and making the behavior between number types // and string consistent.
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/eager/gradients.h
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`. class GradientRegistry { public:
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)