Home
last modified time | relevance | path

Searched defs:r_input (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp902 Tensor r_input = is_batched ? input : input.unsqueeze(0); in forward() local
959 Tensor r_input = is_batched ? input : input.unsqueeze(0); in forward() local
1010 Tensor r_input = is_batched ? input : input.unsqueeze(0); in forward() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dcanonicalize.cpp20 auto* r_input = r->addInput(); in Canonicalize() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/nouveau/nv30/
H A Dnvfx_fragprog.c27 struct nvfx_reg r_input[PIPE_MAX_SHADER_INPUTS]; member