- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetAvgPoolOpPadAttr (0.52 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
// See the function doc in the header file. def GetAvgPoolOpPadAttr: NativeCodeCall<"GetAvgPoolOpPadAttr($_builder, (*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">; // Returns true if the provided padding in the composite op can *not* be // satisfied by SAME or VALID tensorflow padding. def HasCustomPadding:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.h
// attribute to be passed to the a tfl.average_pool_2d that can fully replace // this composite (here, padding is done directly by the tfl.average_pool_2d as // opposed to being extracted into a separate tfl.pad). StringAttr GetAvgPoolOpPadAttr(Builder& builder, mhlo::CompositeOp op); // Get dense attr for a matrix that corrects the over counting of divisors when // casting an average pool with ceil mode on in terms of average pool with it // off.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
auto op_type = mlir::cast<RankedTensorType>(op->getResult(0).getType()); return RankedTensorType::get(shape, op_type.getElementType()); } StringAttr GetAvgPoolOpPadAttr(Builder& builder, CompositeOp op) { const TorchAvgPoolData pool = GetTorchAvgPoolData(op); if (pool.ph == 0 && pool.pw == 0) { return builder.getStringAttr("VALID"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0)