set_autodiff_node()

void syten::STensorImpl::STensorProxy::set_autodiff_node	(	std::string &&	opname_,
		Vec< Pair< Autodiff::ComputeNodePtr, Size > > &&	input_nodes_,
		Vec< Autodiff::STensorId > &&	output_ids,
		Autodiff::AdjointEvaluator &&	func_,
		Vec< STensor > &&	output_shapes_,
		Vec< AsyncCached< STensor > > &&	cached_tensors_ = {}
	)

Constructs a new ComputeNode from the supplied arguments and sets this proxy's shared pointer to point at the newly-constructed node.

Checks that the supplied vector of output ids contains the current tensor ID.

References ad_compute_node, ad_id, syten::STensorImpl::Autodiff::create(), syten::hasMember(), std::move(), and SYTEN_ASSERT_MSG.

Referenced by parity().

Here is the call graph for this function:

Here is the caller graph for this function: