Searched defs:input_permuted (Results 1 – 3 of 3) sorted by relevance
145 TensorInfo input_permuted(src->clone()->set_tensor_shape(permuted_shape)); in validate() local168 …CpuAuxTensorHandler input_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_SRC), _input_permute… in run() local
105 const auto input_permuted = input_reshaped.permute(permutation); in math_pixel_shuffle() local146 const auto input_permuted = input_reshaped.permute(permutation); in math_pixel_unshuffle() local
112 TensorInfo input_permuted(src.clone()->set_tensor_shape(permuted_shape)); in validate() local