- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for restoreInsertionPoint (0.27 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
symbol_table.insert(quantization_func); FlatSymbolRefAttr func_name_attr = FlatSymbolRefAttr::get(rewriter.getStringAttr(func_name)); rewriter.restoreInsertionPoint(original_point); auto quantize_call = rewriter.create<TF::PartitionedCallOp>( quantized_op->getLoc(), quantize_result_type, input, func_name_attr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
builder->create<tf_device::ReturnOp>(loc, op->getResults()); // Move op inside launch. op->moveBefore(launch.GetBody().getTerminator()); builder->restoreInsertionPoint(insert_point); } // Performs the transformation for a replicate op inside a while loop. Returns // true when any change was made by this function. bool HandleReplicateOp(TF::WhileRegionOp while_op,
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/tensorflow/transforms/sparsecore/embedding_sequencing.cc
std::vector<Operation*> topological_order; for (Operation& op : parent_func.getOps()) if (ops.contains(&op)) topological_order.push_back(&op); // Create the partitioned call builder.restoreInsertionPoint(insertion_point); auto caller = MakeFuncCaller(builder, module.getLoc(), new_func, inputs); Block* block = new_func.addEntryBlock();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0)