- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for AttrBuilder (0.18 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/c_api_experimental.h
// type-checking purposes like if the op will run on a particular device type. typedef struct TF_AttrBuilder TF_AttrBuilder; TF_CAPI_EXPORT extern TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name); TF_CAPI_EXPORT extern void TF_DeleteAttrBuilder(TF_AttrBuilder* builder); TF_CAPI_EXPORT extern void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name,
C - Registered: Tue Apr 16 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
struct TF_AttrBuilder : public tensorflow::AttrBuilder { using tensorflow::AttrBuilder::AttrBuilder; // The string buffers to make sure that any `attr_name` we pass into // `builder->Set()` will outlive the subsequent // `TF_AttrBuilderCheckCanRunOnDevice()` call(s) on the same `builder`. std::set<std::string> attr_names; }; TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name) {
C++ - Registered: Tue Apr 16 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K 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/experimental/gradients/math_grad.cc
AbstractTensorHandlePtr sqrt_; }; class MatMulGradientFunction : public GradientFunction { public: explicit MatMulGradientFunction(vector<AbstractTensorHandle*> f_inputs, AttrBuilder f_attrs) : forward_inputs_(f_inputs), forward_attrs_(f_attrs) { for (auto input : forward_inputs_) { if (input) { input->Ref(); } } }
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/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 16 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/gradients.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 Apr 16 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0)