Searched refs:strided_shape (Results 1 – 1 of 1) sorted by relevance
378 ShapeVec strided_shape; in StrideOrInflate() local383 should_inflate ? strided_shape.size() : inflated_shape.size(); in StrideOrInflate()385 strided_shape.push_back(dim); in StrideOrInflate()398 TensorShape(should_inflate ? inflated_shape : strided_shape); in StrideOrInflate()407 auto input_map = input.shaped<T, N>(strided_shape); \ in StrideOrInflate()413 auto output_map = output->shaped<T, N>(strided_shape); \ in StrideOrInflate()