- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 91 for foo_attr (0.21 sec)
-
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/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) -
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) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
// destination type is not inferred and must be given explicitly. // // Preconditions: The given value must have a ShapedType. static Value CreateTFCastOpF32(OpBuilder *builder, Location loc, Value x, BoolAttr truncate) { auto x_type = mlir::dyn_cast_or_null<ShapedType>(x.getType()); if (!x_type) llvm_unreachable("unsupported type"); Type type = x_type.clone(builder->getF32Type());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (2) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td
Note: Additional result corresponds to the control output. }]; let arguments = (ins AnyType:$data, StrAttr:$frame_name, DefaultValuedOptionalAttr<BoolAttr, "false">:$is_constant, DefaultValuedOptionalAttr<I64Attr, "10">:$parallel_iterations, // Optional extra control inputs. Variadic<TfeControlType>:$controlInputs ); let results = (outs
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 23 19:35:12 UTC 2023 - 22K bytes - Viewed (0)