Gradient of gno_kernel_eval w.r.t. kernel_params (right operand)
| Type | Intent | Optional | Attributes | Name | ||
|---|---|---|---|---|---|---|
| class(array_type), | intent(inout) | :: | this |
Forward result node containing saved operands |
||
| type(array_type), | intent(in) | :: | upstream_grad |
Upstream gradient tensor |
Gradient tensor for packed kernel parameters
function get_partial_gno_kernel_params(this, upstream_grad) result(output) !! Gradient of gno_kernel_eval w.r.t. kernel_params (right operand) implicit none ! Arguments class(array_type), intent(inout) :: this !! Forward result node containing saved operands type(array_type), intent(in) :: upstream_grad !! Upstream gradient tensor type(array_type) :: output !! Gradient tensor for packed kernel parameters call output%allocate(array_shape=shape(this%right_operand%val)) call this%get_partial_right_val(upstream_grad%val, output%val) end function get_partial_gno_kernel_params