Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 36 for BoolAttr (0.14 sec)

  1. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

      auto operands = adaptor.getOperands();
      assert(operands.size() == 2 && "equal op has two operands");
      auto ctx = getContext();
      if (operands[0] == operands[1]) return BoolAttr::get(ctx, true);
      return BoolAttr::get(ctx, false);
    }
    
    OpFoldResult ConstOp::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
      (void)operands;
      assert(operands.empty() && "constant has no operands");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc

                              lhs_dot_dimensions_info.FlattenedOutDimensionSize()},
                          llvm::ArrayRef<int64_t>{
                              rhs_dot_dimensions_info.FlattenedOutDimensionSize()});
      BoolAttr false_attr = rewriter.getBoolAttr(false);
      auto matmul = rewriter.create<TFL::BatchMatMulOp>(
          loc, RankedTensorType::get(matmul_shape, result_type.getElementType()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  3. 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)
  4. 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)
  5. 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)
  6. tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc

    bool IsStatefulFunctionalControlFlowOp(Operation* op) {
      if (!isa<TF::IfOp, TF::CaseOp, TF::WhileOp>(op)) {
        return false;
      }
    
      if (auto is_stateless = op->getAttrOfType<BoolAttr>("is_stateless")) {
        return !is_stateless.getValue();
      }
      return false;
    }
    
    // Add control dependencies from stateful control-flow ops to graph fetch op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 16.7K bytes
    - Viewed (0)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top