- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 36 for BoolAttr (0.16 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)