- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 382 for get_operations (0.34 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py
""" output_renaming_map = {} with session.Session(graph=ops.Graph()): importer.import_graph_def(graph_def, name='') graph = ops.get_default_graph() for op in graph.get_operations(): if op.type == '_Retval': expected_node_name = op.name if op.get_attr('tf_saved_model.index_path') is not None: index_path_name = op.get_attr('tf_saved_model.index_path')[0]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_executor_to_functional.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
for (Block& block : region->getBlocks()) { for (Operation& op : block.getOperations()) { if (op.hasTrait<OpTrait::ReturnLike>()) result = &op; } } return result; } void EraseReturnOperands(Region* region, llvm::BitVector& erase) { for (Block& block : region->getBlocks()) { for (Operation& op : block.getOperations()) { if (!op.hasTrait<OpTrait::ReturnLike>()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc
// Move all inner ops of the launch to the block containing the launch. auto body = launch.GetBody().without_terminator(); Operation* launch_op = launch.getOperation(); launch_op->getBlock()->getOperations().splice( launch_op->getIterator(), launch.GetBody().getOperations(), body.begin(), body.end()); launch.erase(); return success(); } void LaunchToDeviceAttributePass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_inline_tpu_island.cc
for (func::FuncOp func_op : llvm::make_early_inc_range(nested_block.getOps<func::FuncOp>())) { if (!symbol_table.lookupSymbolIn(getOperation(), func_op.getName())) { nested_block.getOperations().remove(func_op.getOperation()); symbol_table.insert(func_op.getOperation()); } } nested_module->erase(); } } // namespace std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc
const TF::SideEffectAnalysis::Info& side_effect_analysis) { // IfRegion ops must be in the same block. if (second_if.getOperation()->getBlock() != first_if.getOperation()->getBlock()) { return false; } assert(first_if.getOperation()->isBeforeInBlock(second_if.getOperation())); llvm::SmallSetVector<Operation*, 4> destination_ops = GetAllOpsFromIf(first_if);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_functional_to_executor.cc
to_fetch = to_fetch.drop_back(); } builder.create<tf_executor::FetchOp>(loc, to_fetch); // Build Island. island.getBody().push_back(new Block); island.getBody().front().getOperations().splice( island.getBody().front().begin(), body.getOperations(), copy_range.begin(), copy_range.end()); builder.setInsertionPointToEnd(&island.getBody().front()); builder.create<tf_executor::YieldOp>(loc, return_op.getOperands());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sink_constant.cc
} if (constant.hasOneUse()) { LLVM_DEBUG(llvm::dbgs() << "Moved constant " << constant << "\n"); const_op.getOperation()->moveBefore(&body.begin()->front()); return; } map_entry.first->getSecond() = const_op.clone(); body.begin()->getOperations().insert(body.begin()->begin(), map_entry.first->getSecond());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/materialize_mlir_passthrough_op.cc
llvm::zip(block.getArguments(), op->getOperands())) { std::get<0>(arg_mapping).replaceAllUsesWith(std::get<1>(arg_mapping)); } op->getBlock()->getOperations().splice(op->getIterator(), block.getOperations(), block.begin(), std::prev(block.end())); Operation &return_op = block.front(); for (auto ret_mapping :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
replaceAllUsesInRegionWith(std::get<0>(p), std::get<1>(p), op_region); } // Move all instructions in cluster_op into outlined_function's only block. auto& op_body = op.GetBody().getOperations(); outlined_func_block->getOperations().splice( outlined_func_block->end(), op_body, op_body.begin(), op_body.end()); // Replace `tf_device.return` terminator with `std.return` in function // body. auto return_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0)