- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for AveragePool2DOp (0.18 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
int64_t channel = input_type.getDimSize(3); auto avg_pool_output_type = RankedTensorType::get( {batch, 1, 1, channel}, input_type.getElementType()); auto avg_pool = rewriter.create<TFL::AveragePool2DOp>( mean_op.getLoc(), avg_pool_output_type, input, rewriter.getI32IntegerAttr(height), rewriter.getI32IntegerAttr(width), rewriter.getStringAttr("VALID"), rewriter.getI32IntegerAttr(1),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0)