Searched defs:r_input (Results 1 – 3 of 3) sorted by relevance
902 Tensor r_input = is_batched ? input : input.unsqueeze(0); in forward() local959 Tensor r_input = is_batched ? input : input.unsqueeze(0); in forward() local1010 Tensor r_input = is_batched ? input : input.unsqueeze(0); in forward() local
20 auto* r_input = r->addInput(); in Canonicalize() local
27 struct nvfx_reg r_input[PIPE_MAX_SHADER_INPUTS]; member