- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for AveragePool2DOp (0.23 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) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
//===----------------------------------------------------------------------===// // AveragePool2DOp //===----------------------------------------------------------------------===// int64_t AveragePool2DOp::GetArithmeticCount(Operation* op) { int64_t count; if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) { auto avg_pool = llvm::dyn_cast<AveragePool2DOp>(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)