Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 63 for BoolAttr (0.12 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

                                          ImplicitLocOpBuilder& builder,
                                          bool is_lhs) {
      auto operand_type = mlir::cast<ShapedType>(operand.getType());
      BoolAttr true_attr = builder.getBoolAttr(true);
      auto operand_shape = builder.create<TF::ShapeOp>(operand, true_attr);
      const int64_t operand_rank = operand_type.getRank();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc

               << "]";
    
      return success();
    }
    
    bool VariableIsInitialized(TF::VarHandleOp var_handle_op) {
      auto is_variable_initialized =
          var_handle_op->getAttrOfType<BoolAttr>("_is_initialized");
      // Assume variable is initialized if attribute is not set.
      // There are paths that doesn't mark the variables. All variables
      // that doesn't have the attribute will be promoted.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td

      NativeCodeCall<"GetI64ElementsAttr({" # value # "}, &$_builder)">;
    
    class GetF32Scalar<int value> :
      NativeCodeCall<"GetF32Scalar(&$_builder, " # value # ")">;
    
    def TrueBoolAttr : AttrConstraint<CPred<"$_self.cast<::mlir::BoolAttr>().getValue()">>;
    
    def CreateTFShapeOp : NativeCodeCall<
        "$_builder.create<TF::ShapeOp>($0.getLoc(), $1, $2)">;
    
    def IsI32 : NativeCodeCall<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 24.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc

        return true;
      }
    
      OpBuilder builder(func);
      bool is_signed = quant_specs_.IsSignedInferenceType();
      IntegerAttr num_bits =
          builder.getI32IntegerAttr(quant_specs_.GetQuantizationTypeWidth());
      BoolAttr narrow_range = builder.getBoolAttr(false);
    
      auto add_quantize_op = [&](Location loc, Type input_type, Block* block,
                                 Block::iterator insertion_point, Value arg,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

        return true;
      }
    
      OpBuilder builder(func);
      bool is_signed = quant_specs_.IsSignedInferenceType();
      IntegerAttr num_bits =
          builder.getI32IntegerAttr(quant_specs_.GetQuantizationTypeWidth());
      BoolAttr narrow_range = builder.getBoolAttr(false);
    
      auto add_quantize_op = [&](Location loc, Type input_type, Block* block,
                                 Block::iterator insertion_point, Value arg,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc

            // the asymmetric_quantize_inputs attribute in the builtin option.
            dynamic_range_quantized_user->setAttr(
                kAsymmetricQuantizeInputsAttr,
                BoolAttr::get(rewriter.getContext(), true));
            changed = true;
          }
        }
        return changed;
      }
    
      // Convert ConstantOp-CastOp-Operation sequence into new ConstantOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

        } else {
          bool disable_call_shape_inference = false;
          if (op->hasAttr("_disable_call_shape_inference")) {
            disable_call_shape_inference =
                op->getAttrOfType<BoolAttr>("_disable_call_shape_inference")
                    .getValue();
          }
          inner_op =
              rewriter.create<LegacyCallOp>(loc, new_types, inner_op_operands,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        // Set to `nullptr` because this attribute only matters when the input is
        // dynamic-range quantized.
        const BoolAttr asymmetric_quantize_inputs = nullptr;
    
        const int lhs_rank = lhs_value.getType().cast<TensorType>().getRank();
        const BoolAttr adj_x =
            (lhs_contracting_dims[0] == lhs_rank - 2 ? rewriter.getBoolAttr(true)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

                        const QuantPassSpec& quant_params) {
      rewriter.setInsertionPointAfter(quantized_op);
      FloatAttr tolerance = rewriter.getF32FloatAttr(
          quant_params.numeric_verify_spec.error_tolerance);
      BoolAttr log =
          rewriter.getBoolAttr(quant_params.numeric_verify_spec.log_if_failed_flag);
      // Verify the quantized value by sending the result to the verifier.
      rewriter.create<VerifierT>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

                        Attribute value) {
      ShapedType type;
      if (auto elem_attr = mlir::dyn_cast<ElementsAttr>(value)) {
        return ConstOp::build(builder, result, elem_attr);
      } else if (mlir::isa<BoolAttr, FloatAttr, IntegerAttr>(value)) {
        // All TensorFlow types must be tensor types. In the build() method,
        // we want to provide more flexibility by allowing attributes of scalar
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
Back to top