Searched refs:jacobian_t (Results 1 – 1 of 1) sorted by relevance
316 Tensor jacobian_t(jacobian_type, {x_size, y_size}); in InitJacobians() local317 auto jacobian_t_flat = jacobian_t.flat<JAC_T>(); in InitJacobians()319 (*jacobians)[x_idx * y_num + y_idx] = std::move(jacobian_t); in InitJacobians()