- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 67 for BoolAttr (0.1 sec)
-
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 (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
return Unimplemented("SetAttrFloat has not been implemented yet."); } Status MlirAbstractOp::SetAttrBool(const char* attr_name, bool value) { attrs_[attr_name] = BoolAttr::get(context_, value); return absl::OkStatus(); } Status MlirAbstractOp::SetAttrShape(const char* attr_name, const int64_t* dims, const int num_dims) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
// F32Attr already returns a float as required by flatbuffer builders. static float ConvertF32AttrForOptionWriter( llvm::APFloat f, flatbuffers::FlatBufferBuilder* builder) { return f.convertToFloat(); } // BoolAttr already returns a bool as required by flatbuffer builders. static bool ConvertBoolAttrForOptionWriter( bool b, flatbuffers::FlatBufferBuilder* builder) { return b; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
TF::TensorListLengthOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { Location loc = op.getLoc(); Value input_handle = adaptor.getOperands()[0]; BoolAttr true_attr = rewriter.getBoolAttr(true); auto shape = rewriter.create<TF::ShapeOp>(loc, input_handle, /*use_32bit=*/true_attr); rewriter.replaceOpWithNewOp<TF::GatherOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
}; TF::ReshapeOp createOutputReshapeOpForDynamic( Value value, ArrayRef<int64_t> shape, Value org_lhs, Value org_rhs, EinsumDimensionNumbers& dnums, Location loc, PatternRewriter* rewriter) { BoolAttr true_attr = rewriter->getBoolAttr(true); // Build ShapeOp auto shape_lhs = rewriter->create<TF::ShapeOp>(loc, org_lhs, true_attr); auto shape_rhs = rewriter->create<TF::ShapeOp>(loc, org_rhs, true_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0)