Searched defs:sample_output (Results 1 – 2 of 2) sorted by relevance
110 auto sample_output = output_list.get(i).toTensor(); in preprocess() local
877 vjp_fn, sample_output argument904 vjp_fn, sample_output, v argument