- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 127 for getLoc (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
ModuleOp module = getOperation(); module.walk([&](tf_device::LaunchFuncOp launch) { OpBuilder builder(launch); auto call_op = builder.create<TF::PartitionedCallOp>( module.getLoc(), launch.getResultTypes(), launch.getOperands(), SymbolRefAttr::get(builder.getContext(), launch.getFunc()), /*config=*/builder.getStringAttr(""), /*config_proto=*/builder.getStringAttr(""),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
SmallVector<NamedAttribute> attrs{op->getAttrs()}; attrs.push_back(call_target); auto custom_call = rewriter.create<mlir::stablehlo::CustomCallOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), attrs); rewriter.replaceOp(op, custom_call.getResults()); return success(); } } // namespace template <typename OpTy>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
} TypeAttr type_attr = TypeAttr::get(new_type); auto quantize = builder.create<TFL::QuantizeOp>(value.getLoc(), new_type, value, type_attr); auto dequantize = builder.create<TFL::DequantizeOp>( value.getLoc(), expressed_type, quantize.getOutput()); value.replaceAllUsesWith(dequantize);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_custom_call_to_composite.cc
auto decomposition = mlir::cast<FlatSymbolRefAttr>(calledComputations[0]); auto composite = rewriter.create<mlir::stablehlo::CompositeOp>( op.getLoc(), op.getResultTypes(), op.getOperands(), name.str(), attrs, decomposition.getValue()); rewriter.replaceOp(op, composite.getResults()); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc
if (!input_op) { return failure(); } Builder builder(op.getContext()); auto new_gather_op = rewriter.create<TFL::GatherOp>( op.getLoc(), /*result=*/ mlir::cast<TensorType>(op.getResult().getType()) .clone(builder.getI4Type()), /*operand=*/op.getOperands(), op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K 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/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/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/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
const auto composite_resource_users = GetResourceOpsUsingCompositeArgsInReplicate(replicate_op); for (auto resource_user : composite_resource_users) { WrapOpInLaunch(builder, resource_user->getLoc(), resource_user, tensorflow::GetDeviceAliasForLogicalCore(0)); } } void TPUColocateCompositeResourceOps::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
if (call_op_has_platform_index_arg) { arg_mapper.map(func_op.getArgument(0), builder.create<mhlo::ConstantOp>( func_op.getLoc(), builder.getI16IntegerAttr(0))); } for (auto [func_arg, operand] : llvm::zip_equal( func_op.getArguments().take_back(call_op.getNumOperands()), call_op.getOperands())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 4.8K bytes - Viewed (0)