- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TPUPartitionedCallOp (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
// a TF operation. bool isLegalToInline(Operation *call, Operation *callable, bool wouldBeCloned) const final { // Skip inlining for TPUPartitionedCalls and RemoteCalls. if (isa<TPUPartitionedCallOp>(call)) return false; if (isa<RemoteCallOp>(call)) return false; // Maintain inlining for `tf.function`s with jit_compile option. if (callable->hasAttr("tf._XlaMustCompile")) return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
new llvm::SmallDenseSet<mlir::TypeID, 32>{ TypeID::get<TF::StatefulPartitionedCallOp>(), TypeID::get<TF::PartitionedCallOp>(), TypeID::get<TF::TPUPartitionedCallOp>(), }; return ops_set; }(); auto abstractOp = op->getRegisteredInfo(); if (!abstractOp) return false; return ops->count(abstractOp->getTypeID()) != 0; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
for (auto func : module.getOps<func::FuncOp>()) { for (auto user : symbol_map.getUsers(func)) { // Populates work queue with func ops called from TPUPartionedCall. if (llvm::isa<TF::TPUPartitionedCallOp>(user)) { function_worklist.push(func); } // Populates caller to called func map. if (func::FuncOp caller = user->getParentOfType<func::FuncOp>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0)