Searched defs:workspaceI (Results 1 – 1 of 1) sorted by relevance
388 auto workspaceI = getTensorInfo<scalar_t, index_type>(workspace); in lstm_forward_impl() local422 auto workspaceI = getTensorInfo<scalar_t, index_type>(workspace); in lstm_backward_impl() local460 auto workspaceI = getTensorInfo<scalar_t, index_type>(workspace); in gru_forward_impl() local489 auto workspaceI = getTensorInfo<scalar_t, index_type>(workspace); in gru_backward_impl() local