#[unsafe(no_mangle)]pub unsafe extern "C" fn tfe_tropical_einsum_rrule_minplus_f64(
subscripts: *const c_char,
operands: *const *const TfeTensorF64,
num_operands: usize,
cotangent: *const TfeTensorF64,
grads_out: *mut *mut TfeTensorF64,
status: *mut tfe_status_t,
)Expand description
Reverse-mode rule (VJP) for MinPlus tropical einsum.
Computes one gradient tensor per input operand given the output cotangent.
§Safety
subscriptsmust be a valid null-terminated C string.operandsmust point to an array ofnum_operandsvalid tensor pointers.cotangentmust be a valid, non-null tensor pointer.grads_outmust point to a caller-allocated array ofnum_operandsmutable output pointers.statusmust be a valid, non-null pointer.
§Examples (C)
tfe_tensor_f64 *grads[2];
tfe_status_t status;
const tfe_tensor_f64 *ops[] = {a, b};
tfe_tropical_einsum_rrule_minplus_f64("ij,jk->ik", ops, 2, grad_c, grads, &status);
tfe_tensor_f64_release(grads[0]);
tfe_tensor_f64_release(grads[1]);