- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 46 for setInsertionPointAfter (0.21 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc
rewriter.setInsertionPoint(op); WrapOperandsInUnrealizedCastAndConvert(op, converter, rewriter); // Convert op types for (auto value : op->getResults()) { rewriter.setInsertionPointAfter(value.getDefiningOp()); ConvertAndWrapUsesInUnrealizedCast(value, converter, rewriter); } // Convert block arguments for (auto ®ion : op->getRegions()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 19:48:51 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
if (!host_module.empty()) { mlir::OwningOpRef<mlir::ModuleOp> module_for_func; func::FuncOp func = op.GetHostFunc(&module_for_func); OpBuilder::InsertionGuard guard(rewriter); rewriter.setInsertionPointAfter(op->getParentOfType<func::FuncOp>()); cloned_func = llvm::dyn_cast_or_null<func::FuncOp>( rewriter.clone(*func.getOperation())); manager.insert(cloned_func);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
func::FuncOp CreateInitializerFunc(ModuleOp module_op) { SessionInitializerOp session_init_op = GetOrCreateSessionInitializerOp(module_op); OpBuilder builder(module_op.getContext()); builder.setInsertionPointAfter(session_init_op); const Location loc = builder.getUnknownLoc(); const auto func_type = builder.getFunctionType(/*inputs=*/{}, /*results=*/{}); auto init_func = builder.create<func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
<< ", loc: " << owning_op->getLoc() << "\n"); continue; } mlir::OpBuilder builder{owning_op->getContext()}; builder.setInsertionPointAfter(defining_op); auto const_op_cloned = builder.clone(*defining_op); // Replace the operand with the duplicated op. owning_op->setOperand(curr_operand->getOperandNumber(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
break; } if (has_external_uses) { launch_results.push_back(result); launch_result_types.push_back(result.getType()); } } } builder->setInsertionPointAfter(before_op); auto launch = builder->create<tf_device::LaunchOp>( before_op->getLoc(), builder->getStringAttr(host_device), launch_result_types); launch.getBody().push_back(launch_block);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
if (has_external_uses) { launch_results.push_back(result); launch_result_types.push_back(result.getType()); } } } before ? builder->setInsertionPoint(op) : builder->setInsertionPointAfter(op); auto launch = builder->create<mlir::tf_device::LaunchOp>( op->getLoc(), builder->getStringAttr(host_device), launch_result_types); launch.getBody().push_back(launch_block);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
state_var_vals.front().getType()); // Build a constant default key to specify that the unformatting should // transform the variables to the original format. builder.setInsertionPointAfter(while_op); tensorflow::Tensor default_key_tensor(tensorflow::DT_STRING, {3}); default_key_tensor.vec<tensorflow::tstring>()(0) = kDefaultShardingValue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
const SmallVectorImpl<Value>& inputs, const NamedAttrList& attr_list, const llvm::StringMap<Attribute>& derived_attrs) const { // Create the new op Location loc = call_op.getLoc(); rewriter.setInsertionPointAfter(call_op); std::string tf_op_name = GetTFOpName(call_op.getCallee()); OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list); Operation* new_op = rewriter.create(new_state);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
Type read_type = read_variable_op.getValue().getType(); if (read_type != last_store.getValue().getType()) { OpBuilder builder(last_store); builder.setInsertionPointAfter(last_store); auto cast = builder.create<TF::CastOp>( last_store.getLoc(), read_type, last_store.getValue(), /*Truncate=*/builder.getBoolAttr(false));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0)