- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for saveInsertionPoint (1.03 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc
mlir::Operation* op, llvm::StringRef device) { mlir::OpBuilder::InsertPoint insert_point = builder->saveInsertionPoint(); auto launch = builder->create<mlir::tf_device::LaunchOp>( loc, builder->getStringAttr(device), op->getResultTypes()); launch.getBody().push_back(new mlir::Block);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.pyi
def create(self, *args, **kwargs) -> Any: ... def getContext(self) -> MLIRContext: ... def getUnknownLoc(self) -> Location: ... def restoreInsertionPoint(self, arg0) -> None: ... def saveInsertionPoint(self, *args, **kwargs) -> Any: ... def setInsertionPoint(self, arg0: Block, arg1: Block_Iterator) -> None: ... class OpBuilder_InsertionPoint: def __init__(self, *args, **kwargs) -> None: ...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 09 17:10:09 UTC 2023 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
QuantizedType)> quantization_operations_func) { Operation* input_op = input_val.getDefiningOp(); auto original_point = rewriter.saveInsertionPoint(); auto unique_func_name = func_name.str(); SymbolTable symbol_table(input_op->getParentOfType<ModuleOp>()); while (symbol_table.lookup(unique_func_name)) { absl::StrAppend(&unique_func_name, "_");
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/sparsecore/embedding_pipelining.cc
// Note, 'inputs' and 'outputs' are the predetermined set of values that // should become the operands and return values, respectively. auto saved_insertion_point = builder.saveInsertionPoint(); func::FuncOp new_func = CreateFnWithSignature(module, symbol_table, inputs, outputs, name); // This preserves the order of the ops that was in the original parent
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
void WrapOpInLaunch(OpBuilder* builder, Location loc, Operation* op, llvm::StringRef device) { OpBuilder::InsertPoint insert_point = builder->saveInsertionPoint(); auto launch = builder->create<tf_device::LaunchOp>( loc, builder->getStringAttr(device), op->getResultTypes()); launch.getBody().push_back(new Block);
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
// Note, 'inputs' and 'outputs' are the predetermined set of values that // should become the operands and return values, respectively. auto insertion_point = builder.saveInsertionPoint(); func::FuncOp new_func = CreateFnWithSignature(module, inputs, outputs, absl::StrCat("_func_", name));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// Create functions in a higher scope before restoring the insertion point. // Additionally, create the SymbolTable before further modifying the module. auto original_point = rewriter.saveInsertionPoint(); rewriter.setInsertionPointAfter(op->getParentOfType<func::FuncOp>()); SymbolTable manager(op->getParentOfType<ModuleOp>()); // Constructs `then_branch`, which is executed when `if_cond` evaluates to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
if (maybe_control_node == control_nodes->end()) { return op->getResults(); } mlir::Region region; region.push_back(new mlir::Block); auto saved_pos = op_builder.saveInsertionPoint(); op_builder.setInsertionPointToEnd(®ion.front()); mlir::Operation* cloned_op = op_builder.clone(*op); // Add the yield operation. op_builder.create<mlir::TFL::YieldOp>(op_loc, cloned_op->getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)