Searched defs:params_flat (Results 1 – 3 of 3) sorted by relevance
165 auto params_flat = params.shaped<T, 4>( in Compute() local173 auto params_flat = in Compute() local
119 auto params_flat = params.flat_outer_dims<T>(); in DoCompute() local
693 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); in Compute() local905 auto params_flat = params->flat_outer_dims<T>(); in DoCompute() local