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()
2103 const std::vector<int>& extended_input_dims = extended_input_shape.dims(); in ShuffleArrayTemplate() local2109 input_strides[2] = extended_input_dims[3]; in ShuffleArrayTemplate()2110 input_strides[1] = input_strides[2] * extended_input_dims[2]; in ShuffleArrayTemplate()2111 input_strides[0] = input_strides[1] * extended_input_dims[1]; in ShuffleArrayTemplate()