Searched defs:out_shaped (Results 1 – 2 of 2) sorted by relevance
112 auto out_shaped = output->shaped<T, 2>({nodes * batch, depth}); in launch() local407 auto out_shaped = output->shaped<T, 2>({nodes * batch, depth}); in launch() local
248 auto out_shaped = output_dnn_data->shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() local559 auto out_shaped = output_dnn_data->shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() local