pytc.solver.isdf_xtc_ccsd.compiled_partial_x_left_memory¶
- pytc.solver.isdf_xtc_ccsd.compiled_partial_x_left_memory(t2, left_out, left_inner, x, *, occupied_pair_batch_size=8, rank_panel_size=128)[source]¶
Return XLA accounting for
P[a,m] P[c,m] X[b,d,m].- Parameters:
t2 (jax.Array)
left_out (jax.Array)
left_inner (jax.Array)
x (jax.Array)
occupied_pair_batch_size (int)
rank_panel_size (int)
- Return type: