Searched defs:in_shaped (Results 1 – 2 of 2) sorted by relevance
105 auto in_shaped = in.shaped<T, 2>({nodes * batch, depth}); in launch() local404 auto in_shaped = in_image.shaped<T, 2>({nodes * batch, depth}); in launch() local
225 auto in_shaped = input.shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() local548 auto in_shaped = orig_input_tensor.shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() local