- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 58 for GetBoolAttr (0.23 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/mark_functions_noinline.cc
// Adds the `tf._noinline = true` attribute to the function if the name // matches. if (noinline_functions.contains(func_op.getSymName())) { func_op->setAttr(kTfNoinlineAttr, builder.getBoolAttr(true)); LLVM_DEBUG(llvm::dbgs() << "Marked tf._noinline = true: " << func_op.getSymName()); } } static PassRegistration<MarkFunctionsNoinlinePass> pass{}; } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 18 02:52:57 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/annotate_parameter_replication.cc
&replicate.getBody())) { // Not a replication-invariant operand. continue; } func.setArgAttr(entry.index(), kReplicationAttr, builder.getBoolAttr(true)); } }); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateAnnotateParameterReplicationPass() { return std::make_unique<AnnotateParameterReplicationPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
// in TensorFlow and allows lowering to V1 control flow for loop // parallelization. dataset_while->setAttr("_lower_using_switch_merge", builder.getBoolAttr(true)); return dataset_while; } // Populate the cond of `dataset_while`. The cond body just returns the // condition of whether to continue to next iteration.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
OpBuilder builder(&ctx); module_op.walk([&builder](Operation* op) { if (op->hasAttrOfType<BoolAttr>("is_stateless") && ContainCalibrationStatisticsSaverOp(op)) { op->setAttr("is_stateless", builder.getBoolAttr(false)); } }); } std::unique_ptr<OperationPass<ModuleOp>> CreateInsertCalibrationStatisticsSaverPass( StringRef calibration_data_dir,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
auto matmul_type = RankedTensorType::get(matmul_shape, original_type.getElementType()); Value out = rewriter.create<TF::BatchMatMulV2Op>( op.getLoc(), matmul_type, lhs, rhs, rewriter.getBoolAttr(false), rewriter.getBoolAttr(false)); bool out_reshape_need = (reshape_shape.size() != matmul_shape.size() || original_type.getRank() != matmul_shape.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/fake_quant_utils.h
rewriter.setInsertionPointAfter(tf_op.getOperation()); IntegerAttr num_bits = rewriter.getI64IntegerAttr(tf_op.getNumBits()); BoolAttr narrow_range = rewriter.getBoolAttr(tf_op.getNarrowRange()); Type res_type = tf_op.getType(); TypeAttr qtype = quant::GetQuantizedTypeAttr( rewriter, input_type, min_value, max_value, quant_dim, num_bits,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td
def convertIntAttrTo32Bit : NativeCodeCall< "$_builder.getI32IntegerAttr($0.cast<IntegerAttr>().getInt())">; // Builds a constant bool attribute. class GetBoolAttr<int value> : NativeCodeCall<"$_builder.getBoolAttr(" # value #")">; // Converts an integer attribute $0 to 64-bit with builder. def convertIntAttrTo64Bit : NativeCodeCall< "$_builder.getI64IntegerAttr($0.cast<IntegerAttr>().getInt())">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 28.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/fake_quant_utils.h
rewriter.setInsertionPointAfter(tf_op.getOperation()); IntegerAttr num_bits = rewriter.getI64IntegerAttr(tf_op.getNumBits()); BoolAttr narrow_range = rewriter.getBoolAttr(tf_op.getNarrowRange()); Type res_type = tf_op.getType(); TypeAttr qtype = quant::GetQuantizedTypeAttr( rewriter, res_type, min_value, max_value, quant_dim, num_bits,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
rewriter.setInsertionPointAfter(tf_op.getOperation()); IntegerAttr num_bits = rewriter.getI64IntegerAttr(tf_op.getNumBits()); BoolAttr narrow_range = rewriter.getBoolAttr(tf_op.getNarrowRange()); Type res_type = tf_op.getType(); TypeAttr qtype = quant::GetQuantizedTypeAttr( rewriter, res_type, min_value, max_value, quant_dim, num_bits,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/ops.cc
[](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, mlir::Value y) -> mlir::Operation* { return opb .create<mlir::TF::EqualOp>(loc, x, y, opb.getBoolAttr(true)) .getOperation(); }); // mlir::TF::GreaterEqualOp py::class_<mlir::TF::GreaterEqualOp>(m, "Tf_GreaterEqualOp") .def("create",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 7.6K bytes - Viewed (0)