Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 31 for MulOp (0.05 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top