- Sort Score
- Result 10 results
- Languages All
Results 171 - 180 of 380 for getLoc (0.26 sec)
-
tensorflow/compiler/mlir/lite/transforms/raise_custom_ops.cc
(target_op_names.empty() && !op->isRegistered())) { custom_ops.push_back(op); } }); for (auto *op : custom_ops) { builder.setInsertionPoint(op); Location loc = op->getLoc(); auto custom_op = builder.create<CustomTfOp>(loc, op->getResultTypes(), op->getOperands()); Region region; Block *new_block = new Block;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc
return failure(); } results.clear(); builder.setInsertionPointAfter(op); for (const auto& result_value : result_values) { results.push_back(builder.create<TF::ConstOp>(op->getLoc(), result_value)); } return success(); } bool IsOperationFoldable(Operation* op) { if (isa<TF::ConstOp>(op)) return true; if (op->getDialect()->getNamespace() != "tf" || !TF::CanBeFolded(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
if (args_permutation.empty() || res_permutation.empty()) return; mlir::Operation* op = layout_sensitive_interface.getOperation(); Location loc = op->getLoc(); OpBuilder builder = OpBuilder::atBlockEnd(op->getBlock()); auto perm_attr = [&](Permutation permutation) -> DenseIntElementsAttr { auto perm_ty = RankedTensorType::get({4}, builder.getIntegerType(64));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_cfg.cc
// completely from the IR, breaking it into operations to evaluate the condition // as a bool, plus some branches. static LogicalResult LowerIfOp(IfOp op) { Operation* op_inst = op.getOperation(); Location loc = op_inst->getLoc(); OpBuilder builder(op_inst); // Lower the condition to a boolean value (i1). Value cond_i1 = LowerCondition(loc, op.getCond(), &builder); if (!cond_i1) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 13 11:42:59 UTC 2023 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
if (!data_format_is_valid) { return emitOptionalError(op.getLoc(), "Invalid data format provided"); } const StringRef paddings = op.getPadding(); tensorflow::Padding padding; auto padding_is_valid = GetPaddingFromString(paddings.str(), &padding); if (!padding_is_valid.ok()) { return emitOptionalError(op.getLoc(), "Invalid padding format provided"); } // Verifies that,
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/utils/xla_rewrite_util_test.cc
std::string device = "/job:localhost/replica:0/task:0/device:CPU:0"; module->walk( [&](mlir::tf_device::ClusterOp descendant) { cluster = descendant; }); mlir::OpBuilder builder(&context); auto loc = cluster->getLoc(); // Wrap the cluster op into a Launch op auto launch_op = tensorflow::WrapOpInLaunch(&builder, loc, cluster, device); EXPECT_TRUE(llvm::isa<mlir::tf_device::LaunchOp>(launch_op)); launch_op->erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
per_axis_type.getStorageTypeMin(), per_axis_type.getStorageTypeMax()); } auto quantize = builder.create<quantfork::QuantizeCastOp>( q_op.getLoc(), new_value_type.clone(new_qtype), new_value); auto dequantize = builder.create<quantfork::DequantizeCastOp>( dq_op.getLoc(), new_value_type, quantize.getResult()); return dequantize.getResult(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
tf_device::LaunchOp compile_launch, OpBuilder* builder) { return builder->create<TF::TPUGetLayoutOp>( compile_launch.getLoc(), llvm::ArrayRef<Type>{RankedTensorType::get({ShapedType::kDynamic}, builder->getIntegerType(64))}, llvm::ArrayRef<Value>{compilation_key},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc
// This occurs when doing qi16 convolution, as bias is passed as a // non-quantized int64 if (allTypesFp || allTypesQuantizedOrInt) return failure(); Location loc = op->getLoc(); SmallVector<Value> newOperands; newOperands.reserve(op->getNumOperands()); for (auto operand : op->getOperands()) { if (QuantizedType::getQuantizedElementType(operand.getType())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_custom_aggregation_op_to_quant_stats.cc
static_cast<float>(max.getValueAsDouble())}); ElementsAttr axis_stats; IntegerAttr axis; quantfork::StatisticsOp stats_op = rewriter.create<quantfork::StatisticsOp>( op->getLoc(), op.getInput(), layer_stats, axis_stats, axis); op.getOutput().replaceAllUsesWith(stats_op.getResult()); return success(); } }; static PassRegistration<ConvertCustomAggregationOpToQuantStatsPass> pass;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.8K bytes - Viewed (0)