Searched refs:extended_input_dims (Results 1 – 2 of 2) sorted by relevance
55 const std::vector<int>& extended_input_dims = extended_input_shape.dims(); in Transpose() local61 input_strides[2] = extended_input_dims[3]; in Transpose()62 input_strides[1] = input_strides[2] * extended_input_dims[2]; in Transpose()63 input_strides[0] = input_strides[1] * extended_input_dims[1]; in Transpose()
2147 const std::vector<int>& extended_input_dims = extended_input_shape.dims(); in ShuffleArrayTemplate() local2153 input_strides[2] = extended_input_dims[3]; in ShuffleArrayTemplate()2154 input_strides[1] = input_strides[2] * extended_input_dims[2]; in ShuffleArrayTemplate()2155 input_strides[0] = input_strides[1] * extended_input_dims[1]; in ShuffleArrayTemplate()