- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetPadOpAttr (0.28 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
// by SAME or VALID tensorflow padding. def HasSameOrValidPadding: Constraint<Neg<HasCustomPadding.predicate>>; // See the function doc in the header file. def GetPadOpAttr: NativeCodeCall<"GetPadOpAttr($_builder, (*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">; // See the function doc in the header file.
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
namespace odml { // Given a Composite op that wraps a core.aten.avg_pool2d, returns the padding // configuration required for the `tfl.pad` if the padding part of the op is // to be done before average pooling. DenseIntElementsAttr GetPadOpAttr(Builder& builder, mhlo::CompositeOp op); // Given a Composite op that wraps a core.aten.avg_pool2d, and assuming that // the padding part is extracted into a tfl.pad op prior to a
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
const int ceil_pad_right = pool.kw - remaining_right; pad_right = ceil_pad_right - pool.pw; } return {pool.ph, pad_bottom, pool.pw, pad_right}; } DenseIntElementsAttr GetPadOpAttr(Builder& builder, CompositeOp op) { const TorchAvgPoolData pool = GetTorchAvgPoolData(op); const auto values = GetPadOpPaddingValues(pool); llvm::SmallVector<int32_t> padding_vec(8, 0); // NHWC
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0)