Searched refs:paddings_array_shape (Results 1 – 1 of 1) sorted by relevance
1287 const auto& paddings_array_shape = paddings_array.shape(); in ProcessSpaceToBatchNDOperator() local1289 QCHECK_EQ(paddings_array_shape.dimensions_count(), 2); in ProcessSpaceToBatchNDOperator()1302 QCHECK_EQ(paddings_array_shape.dims(0), 2); // Number of block dimensions in ProcessSpaceToBatchNDOperator()1303 QCHECK_EQ(paddings_array_shape.dims(1), 2); // Two parameters per dimension. in ProcessSpaceToBatchNDOperator()