- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 230 for GetOperation (0.43 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
if (!op.isPrivate()) return; auto call = llvm::dyn_cast<CallableOpInterface>(op.getOperation()); if (!call) return; Region* region = call.getCallableRegion(); if (!region) return; // happens e.g. for external functions auto func = llvm::dyn_cast<FunctionOpInterface>(op.getOperation()); if (!func || do_not_touch.count(func)) return; llvm::BitVector unused_args(func.getNumArguments());
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/ir/tf_ops_layout_helper.h
// Update types for all layout sensitive results. auto layout_sensitive = cast<LayoutSensitiveInterface>(op->getOperation()); for (unsigned idx : layout_sensitive.GetLayoutDependentResults()) { OpResult result = op->getOperation()->getResult(idx); result.setType(ShuffleRankedTensorType(result.getType(), perm)); } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
func::FuncOp callee = from.lookup<func::FuncOp>(symbol_ref.getValue()); callee.getOperation()->getBlock()->getOperations().remove( callee.getOperation()); to.insert(callee); } void TPUBridgeExecutorIslandOutlining::runOnOperation() { MLIRContext *ctx = &getContext(); SymbolTable symbol_table(getOperation()); if (Operation *nested_module = symbol_table.lookup(kNestedModule)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/device_index_selector.cc
: public impl::DeviceIndexSelectorPassBase<DeviceIndexSelector> { void runOnOperation() override; }; } // namespace void DeviceIndexSelector::runOnOperation() { func::FuncOp func = getOperation(); // Convert all the DeviceIndex ops to constant values. func.getBody().walk([](TF::DeviceIndexOp op) { // This just selects the default in all cases where DeviceIndex feeds into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
if (resource_handle_op == parallel_execute) return nullptr; if (resource_handle_op && resource_handle_op->getBlock() == parallel_execute.getOperation()->getBlock() && parallel_execute.getOperation()->isBeforeInBlock(resource_handle_op)) return nullptr; return assign_var; } // Finds AssignVariableOps that can be moved into a parallel_execute region and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0) -
staging/src/k8s.io/apiserver/pkg/admission/plugin/namespace/lifecycle/admission.go
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Sep 15 09:52:18 UTC 2021 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sink_constant.cc
llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter) : filter_(filter) {} void runOnOperation() override { getOperation().walk([filter = filter_](tf_device::ClusterOp cluster) { LLVM_DEBUG(llvm::dbgs() << "Visit " << *cluster.getOperation() << "\n"); // For each launch op, we find the values used that come from a constant // defined above and sink these constants in the region body.
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/init_text_file_to_import_test_pass.cc
return "generate a temporary file and invoke InitTextFileToImportPass"; } private: void runOnOperation() override; }; void InitTextFileToImportTestPass::runOnOperation() { ModuleOp module = getOperation(); // Create a temporary vocab file. int fd; SmallString<256> filename; std::error_code error_code = llvm::sys::fs::createTemporaryFile("text", "vocab", fd, filename);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_tensor_list_init_ops_pass.cc
std::vector<OpOperand*> uses; for (auto& use : tensor_list.getUses()) { uses.emplace_back(&use); } OpBuilder builder(tensor_list_op.getOperation()); for (OpOperand* operand : uses) { auto new_op = builder.clone(*tensor_list_op.getOperation()); operand->set(new_op->getResult(0)); } } // This transformation pass replicates TensorList initialization ops. class ReplicateTensorListInitOps
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 22 17:28:34 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables_test_pass.cc
public: LiftVariablesTestPass() { session_ = new TF::test_util::FakeSession(); } ~LiftVariablesTestPass() override { delete session_; } void runOnOperation() override { ModuleOp module = getOperation(); if (failed(tf_saved_model::LiftVariables(module, session_))) signalPassFailure(); } private: Session* session_; }; #define GEN_PASS_DEF_LIFTVARIABLESINVALIDSESSIONTESTPASS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 2.7K bytes - Viewed (0)