- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 31 for MulOp (0.05 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
if (!dq_op) { auto mul_op = builder.create<TF::MulOp>(loc, value, multiplier); return mul_op.getResult(); } auto q_op = dq_op.getArg().getDefiningOp<quantfork::QuantizeCastOp>(); if (!q_op) return {}; Value float_value = q_op.getArg(); Value new_value = builder.create<TF::MulOp>(loc, float_value, multiplier); auto new_value_type = mlir::cast<TensorType>(new_value.getType());
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/experimental/tac/hardwares/cpu_hardware.cc
TAC_REGISTER_CPU_OP(Conv2DOp, CreateConvOp); TAC_REGISTER_CPU_OP(DepthwiseConv2DOp, CreateConvOp); TAC_REGISTER_CPU_OP(FullyConnectedOp, CreateConvOp); TAC_REGISTER_CPU_OP(AddOp, CreateArithmeticOp); TAC_REGISTER_CPU_OP(MulOp, CreateArithmeticOp); TAC_REGISTER_CPU_OP(ConcatenationOp, CreateConcatOp); TAC_REGISTER_CPU_OP(ReshapeOp, CreateConcatOp); TAC_REGISTER_CPU_OP(PackOp, CreateConcatOp); #undef TAC_REGISTER_CPU_OP } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
LLVM_DEBUG(llvm::dbgs() << "Failed to match other_input_zero_points_constant_op.\n"); return failure(); } auto combined_scale_multiply_op = dyn_cast_or_null<stablehlo::MulOp>( *subtract_op.getResult().user_begin()); if (!combined_scale_multiply_op) { LLVM_DEBUG(llvm::dbgs() << "Failed to match combined_scale_multiply_op.\n"); return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
tblgen_attrs); } ::mlir::TF::MulOp multiplier; { ::mlir::Value tblgen_value_0 = (*scale.begin()); ::mlir::Value tblgen_value_1 = (*rsqrt_op.getODSResults(0).begin()); multiplier = rewriter.create<::mlir::TF::MulOp>(odsLoc, /*x=*/tblgen_value_0,
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/lite/experimental/tac/hardwares/gpu_hardware.cc
TAC_REGISTER_GPU_OP(Conv2DOp, CreateConvOp); TAC_REGISTER_GPU_OP(DepthwiseConv2DOp, CreateConvOp); TAC_REGISTER_GPU_OP(FullyConnectedOp, CreateConvOp); TAC_REGISTER_GPU_OP(AddOp, CreateArithmeticOp); TAC_REGISTER_GPU_OP(MulOp, CreateArithmeticOp); #undef TAC_REGISTER_GPU_OP } // namespace } // namespace tac } // namespace TFL
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/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
auto reduced = builder.create<TF::SumOp>(loc, tensor_i32, reduction_indices_value, /*keep_dims=*/builder.getBoolAttr(true)); auto mul_op = builder.create<TF::MulOp>(loc, zp, reduced); SmallVector<Value> folded_results = ConstantFoldOpIfPossible(mul_op); return folded_results.front(); } // Add two contributions, and a zeropoint modification term
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
// type. scale_op = rewriter.create<TF::CastOp>( loc, UnrankedTensorType::get(rewriter.getBF16Type()), scale_op); } auto mul_op = rewriter.create<TF::MulOp>(loc, new_cast_op.getType(), scale_op, new_cast_op); return mul_op; } // Add uniform quantization's dequantization logic.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
auto scalar_add = [&](Value lhs, Value rhs) { return builder.create<TF::AddOp>(loc, int32_scalar_type, lhs, rhs); }; auto scalar_mul = [&](Value lhs, Value rhs) { return builder.create<TF::MulOp>(loc, int32_scalar_type, lhs, rhs); }; auto scalar_sub = [&](Value lhs, Value rhs) { return builder.create<TF::SubOp>(loc, int32_scalar_type, lhs, rhs); }; auto scalar_div = [&](Value lhs, Value rhs) {
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/push_transpose_through_ewise.cc
// definition of Elementwise, all i/o shapes and types must be same-rank // broadcastable and fully static. Consider moving this into attribute later. bool IsElementwise(Operation *op) { if (!(llvm::isa<TFL::AddOp, TFL::MulOp, TFL::DivOp, TFL::SubOp, TFL::MaximumOp, TFL::MinimumOp>(op))) { return false; } auto opr1_type = llvm::dyn_cast_or_null<RankedTensorType>(op->getOperand(0).getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
TF::MaxPoolOp::getOperationName().str(), TF::MaximumOp::getOperationName().str(), TF::MeanOp::getOperationName().str(), TF::MinimumOp::getOperationName().str(), TF::MulOp::getOperationName().str(), TF::PadOp::getOperationName().str(), TF::PadV2Op::getOperationName().str(), TF::Relu6Op::getOperationName().str(), TF::ReluOp::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)