- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for forward_attrs_ (0.34 sec)
-
tensorflow/c/experimental/gradients/nn_grad.cc
}; // 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, absl::Span<AbstractTensorHandle*> grad_inputs) override {
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
*/ AbstractTensorHandle* upstream_grad = grad_outputs[0]; // Get transpose attrs bool t_a; TF_RETURN_IF_ERROR(forward_attrs_.Get("transpose_a", &t_a)); bool t_b; TF_RETURN_IF_ERROR(forward_attrs_.Get("transpose_b", &t_b)); // Conj each input AbstractTensorHandle* conj_output; std::string name = "Conj_A_MatMul_Grad"; TF_RETURN_IF_ERROR(
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)