- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 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.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 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
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)