Lines Matching full:narrow
35 Tensor firstHalf = self.narrow(wrap_dim, 0, selfSize); in TORCH_META_FUNC()
36 Tensor secondHalf = self.narrow(wrap_dim, selfSize, selfSize); in TORCH_META_FUNC()
65 Tensor firstHalf = input.narrow(wrap_dim, 0, inputSize); in glu_backward_cpu_out()
66 Tensor secondHalf = input.narrow(wrap_dim, inputSize, inputSize); in glu_backward_cpu_out()
67 Tensor gradInputfirstHalf = grad_input.narrow(wrap_dim, 0, inputSize); in glu_backward_cpu_out()
68 Tensor gradInputsecondHalf = grad_input.narrow(wrap_dim, inputSize, inputSize); in glu_backward_cpu_out()
96 const auto b = x.narrow(dim, glu_size, glu_size); in glu_jvp()
97 const auto da = dx.narrow(dim, 0, glu_size); in glu_jvp()
98 const auto db = dx.narrow(dim, glu_size, glu_size); in glu_jvp()
121 const auto a = x.narrow(dim, 0, glu_size); in glu_backward_jvp()
122 const auto b = x.narrow(dim, glu_size, glu_size); in glu_backward_jvp()
123 const auto da = dx.narrow(dim, 0, glu_size); in glu_backward_jvp()
124 const auto db = dx.narrow(dim, glu_size, glu_size); in glu_backward_jvp()
126 const auto grad_x_a = grad_x.narrow(dim, 0, glu_size); in glu_backward_jvp()
128 const auto grad_x_b = grad_x.narrow(dim, glu_size, glu_size); in glu_backward_jvp()