- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for AveragePool2DOp (0.28 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/hardwares/gpu_hardware.cc
} }; std::unique_ptr<TargetHardwareOperation> CreateConvOp() { return std::make_unique<GpuConvOp>(); } // Op registrations TAC_REGISTER_GPU_OP(AbsOp, CreateBasicOpNoCost); TAC_REGISTER_GPU_OP(AveragePool2DOp, CreateBasicOpNoCost); TAC_REGISTER_GPU_OP(CosOp, CreateBasicOpNoCost); TAC_REGISTER_GPU_OP(DivOp, CreateBasicOpNoCost); TAC_REGISTER_GPU_OP(ExpOp, CreateBasicOpNoCost);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 7.8K bytes - Viewed (0) -
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)