Searched refs:extended_output_dims (Results 1 – 2 of 2) sorted by relevance
56 const std::vector<int>& extended_output_dims = extended_output_shape.dims(); in Transpose() local69 const int output_size_0 = extended_output_dims[3]; in Transpose()70 const int output_size_1 = extended_output_dims[2]; in Transpose()71 const int output_size_2 = extended_output_dims[1]; in Transpose()72 const int output_size_3 = extended_output_dims[0]; in Transpose()
2104 const std::vector<int>& extended_output_dims = extended_output_shape.dims(); in ShuffleArrayTemplate() local2117 const int output_size_0 = extended_output_dims[3]; in ShuffleArrayTemplate()2118 const int output_size_1 = extended_output_dims[2]; in ShuffleArrayTemplate()2119 const int output_size_2 = extended_output_dims[1]; in ShuffleArrayTemplate()2120 const int output_size_3 = extended_output_dims[0]; in ShuffleArrayTemplate()