- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for partition_ops_ (0.13 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
// remainder of that recursive call. auto extract = [&](const llvm::SetVector<Operation*>& partition_ops) -> void { if (partition_ops.empty()) return; InferenceDeviceType device = GetInferenceDeviceTypeForOp(partition_ops.front()).value(); Subgraph old_subgraph(partition_ops, ++func_count); OpsAdded ops_added; ExtractSubgraphToFunc(old_subgraph, builder, module, ops_added);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
llvm::SmallVector<std::string, 4> result_devices; // The operations to be included in the body of the function. llvm::SmallVector<Operation *, 4> ops; func::FuncOp partition_op; }; // Returns a map that maps the host address to the metadata of the function // for that remote host. The metadata of the function specifies the input
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0)