Searched refs:output_shaped (Results 1 – 2 of 2) sorted by relevance
115 auto output_shaped = output->shaped<T, 2>({before_dim, after_dim}); in Compute() local119 output_shaped, input_reshaped, indices, in Compute()
159 auto output_shaped = output.shaped<T, 2>({size, suffix_dim_size}); in SplitCPU() local164 output_shaped, input_reshaped, in SplitCPU()