Searched refs:extended_input_shape (Results 1 – 2 of 2) sorted by relevance
48 Shape extended_input_shape = input_shape; in Transpose() local49 ExtendShape(&extended_input_shape, 4); in Transpose()55 const std::vector<int>& extended_input_dims = extended_input_shape.dims(); in Transpose()
2096 Shape extended_input_shape = input_shape; in ShuffleArrayTemplate() local2097 ExtendShape(&extended_input_shape, 4); in ShuffleArrayTemplate()2103 const std::vector<int>& extended_input_dims = extended_input_shape.dims(); in ShuffleArrayTemplate()