↓ 1 callersFunctionsave_value_table(
filename: str,
values_path: str,
rows: int = 16,
cols: int = 32,
delimiter: str = ",",
analysis/paper_experiments/swizzle_plot_scripts/table_values_16x32.py:22
↓ 1 callersFunctionsave_value_table(
filename: str,
values_path: str,
rows: int = 16,
cols: int = 64,
delimiter: str = ",",
analysis/paper_experiments/swizzle_plot_scripts/table_values_64x16.py:26
↓ 1 callersMethodset_int8_params(self, q_descale, k_descale, v_descale, p_scale, p_descale)
analysis/baselines/attn/triton_baseline_v02.py:68
↓ 1 callersFunctiontrain_epoch(
model: nn.Module,
data_loader: DataLoader,
loss_fn,
optimizer,
device,
scheduler,
training/bert/tasks.py:217