- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for DepthwiseConv2dNativeOp (0.35 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold_test.cc
EXPECT_TRUE( succeeded(applyPatternsAndFoldGreedily(test_func, std::move(patterns)))); auto depthwise_conv_op = FindOperationOfType<TF::DepthwiseConv2dNativeOp>(test_func); EXPECT_THAT(depthwise_conv_op, NotNull()); // The filter of the DepthwiseConv2dNativeOp is expected to be a constant. EXPECT_TRUE(isa<TF::ConstOp>(depthwise_conv_op.getFilter().getDefiningOp())); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dilated_conv.cc
void IdentifyDilatedConvPass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); patterns.add<ConvertTFDilatedConvOp<TF::Conv2DOp>, ConvertTFDilatedConvOp<TF::DepthwiseConv2dNativeOp>>( &getContext()); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateIdentifyDilatedConvPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 07 21:08:41 UTC 2022 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
TF::ConcatV2Op::getOperationName().str(), TF::Conv2DBackpropInputOp::getOperationName().str(), TF::Conv2DOp::getOperationName().str(), TF::DepthwiseConv2dNativeOp::getOperationName().str(), TF::FusedBatchNormV3Op::getOperationName().str(), TF::GatherV2Op::getOperationName().str(), TF::MatMulOp::getOperationName().str(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc
// TODO: b/289744814 - Refactor to have a single source of truth of TF Quant // specs. absl::flat_hash_set<int> GetQuantizableOperands(Operation* op) { absl::flat_hash_set<int> quantizable_operands; if (isa<TF::DepthwiseConv2dNativeOp, TF::Conv2DOp, TF::Conv3DOp, TF::MatMulOp, TF::BatchMatMulOp>(op)) { quantizable_operands.insert(1); } else if (isa<TF::GatherOp>(op)) { quantizable_operands.insert(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_op_quant_spec.cc
// Supported quantizable ops. return isa<TF::XlaConvV2Op, TF::XlaDotV2Op, TF::MatMulOp, TF::Conv2DOp, TF::GatherOp, TF::GatherV2Op, TF::XlaGatherOp, TF::ResourceGatherOp, TF::DepthwiseConv2dNativeOp, TF::Conv3DOp, TF::BatchMatMulV2Op, TF::EinsumOp>(op); } bool IsOpWithInt8TypeOperand(Operation* op) { return (isa<TF::XlaConvV2Op, TF::XlaDotV2Op, TF::XlaGatherOp, TF::GatherOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
// - Conv2DOp // - Conv3DOp // - DepthwiseConv2dNativeOp // - MatMulOp // - BatchMatMulV2Op LogicalResult MatchSupportedAffineOp(Operation* op, Value& binding_output, Value& binding_input, Value& binding_weight) { bool is_supported_affine_op = false; if (llvm::isa<TF::Conv2DOp, TF::Conv3DOp, TF::DepthwiseConv2dNativeOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
}; class ConvertTFDepthwiseConv2dNative : public ConvertTFConvOp<ConvertTFDepthwiseConv2dNative, TF::DepthwiseConv2dNativeOp> { public: using BaseType = ConvertTFConvOp<ConvertTFDepthwiseConv2dNative, TF::DepthwiseConv2dNativeOp>; ConvertTFDepthwiseConv2dNative(MLIRContext *context, bool allow_bf16_type_legalization)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::Conv3DBackpropFilterV2Op>(), TypeID::get<TF::Conv3DBackpropInputV2Op>(), TypeID::get<TF::CumprodOp>(), TypeID::get<TF::CumsumOp>(), TypeID::get<TF::DepthwiseConv2dNativeOp>(), TypeID::get<TF::DivOp>(), TypeID::get<TF::DynamicStitchOp>(), TypeID::get<TF::_EagerConstOp>(), TypeID::get<TF::EmptyOp>(), TypeID::get<TF::ExpandDimsOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
return false; } } } // All ones in "lhs_dilation" means this "mhlo.conv" op should be // converted to "tf.Conv2D" or "tf.DepthwiseConv2dNativeOp". auto lhs_dilation = conv_op.getLhsDilation().value(); if (!lhs_dilation.isSplat() || lhs_dilation.getSplatValue<int64_t>() != 1) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
}; using ConvertConv2DOp = ConvertConvOp<TF::Conv2DOp, /*num_spatial_dims=*/2>; using ConvertConv3DOp = ConvertConvOp<TF::Conv3DOp, /*num_spatial_dims=*/3>; using ConvertDepthConv2DOp = ConvertConvOp<TF::DepthwiseConv2dNativeOp, /*num_spatial_dims=*/2, /*depthwise_conv=*/true>; // Converts tf.PadV2Op to mhlo.DynamicPadOp. Padding values must be const. class ConvertPadOpDynamic : public OpRewritePattern<TF::PadV2Op> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)