Searched refs:crops_array_shape (Results 1 – 1 of 1) sorted by relevance
1336 const auto& crops_array_shape = crops_array.shape(); in ProcessBatchToSpaceNDOperator() local1338 QCHECK_EQ(crops_array_shape.dimensions_count(), 2); in ProcessBatchToSpaceNDOperator()1351 QCHECK_EQ(crops_array_shape.dims(0), 2); // Number of block dimensions in ProcessBatchToSpaceNDOperator()1352 QCHECK_EQ(crops_array_shape.dims(1), 2); // Two parameters per dimension. in ProcessBatchToSpaceNDOperator()