Searched defs:params_flat (Results 1 – 3 of 3) sorted by relevance
105 auto params_flat = in Compute() local
125 auto params_flat = params.flat_outer_dims<T>(); in DoCompute() local213 auto params_flat = params.flat_outer_dims<T>(); in DoCompute() local
689 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); in Compute() local786 auto params_flat = params->flat_outer_dims<T>(); in Compute() local