- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for TPUPartitionedCallOp (0.3 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/inference/inference_metrics_pass.cc
namespace internal { auto* has_tpu_partitioned_call_streamz = tensorflow::monitoring::Counter<1>::New( "/tensorflow/core/tf2xla/internal/inference/tpu_partitioned_call", "Whether the model has TPUPartitionedCallOp.", "has_tpu_partitioned_call"); namespace { #define GEN_PASS_DEF_INFERENCEMETRICSPASS #include "tensorflow/compiler/mlir/tf2xla/internal/inference/inference_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 05 21:24:51 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
} return success(); } }; class ReplaceTpuPartitionedCallOpWithPartitionedCallOp : public OpRewritePattern<TF::TPUPartitionedCallOp> { public: using OpRewritePattern<TF::TPUPartitionedCallOp>::OpRewritePattern; private: LogicalResult matchAndRewrite(TF::TPUPartitionedCallOp call_op, PatternRewriter& rewriter) const override {
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/tests/inlining.mlir
%result = "tf.PartitionedCall"() {config = "", config_proto = "", executor_type = "", f = @simple_callee_with_noinline} : () -> tensor<2xi32> func.return %result : tensor<2xi32> } // Test that TPUPartitionedCallOp is not inlined. // CHECK-LABEL: func @dont_inline_tpu_partitioned_call( func.func @dont_inline_tpu_partitioned_call() -> tensor<2xi32> { // CHECK-NEXT: %[[ORDINAL:.*]] = "tf.TPUOrdinalSelector"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
} bool HasTPUPartitionedCallOpInModule(mlir::ModuleOp module) { bool has_tpu_partitioned_call = false; for (auto func_op : module.getOps<mlir::func::FuncOp>()) { func_op->walk([&](mlir::TF::TPUPartitionedCallOp op) { has_tpu_partitioned_call = true; }); if (has_tpu_partitioned_call) break; } return has_tpu_partitioned_call; } bool IsInferenceGraph(const Graph& graph,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
} LogicalResult StatefulPartitionedCallOp::verifySymbolUses( SymbolTableCollection &symbolTable) { return VerifyPartitionedCall(*this, symbolTable); } LogicalResult TPUPartitionedCallOp::verifySymbolUses( SymbolTableCollection &symbolTable) { return VerifyPartitionedCall(*this, symbolTable); } template <typename CallOpClass> static void SetPartitionCalleeFromCallable(CallOpClass op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
if (!llvm::hasSingleElement(callers) && !llvm::all_of(callers.drop_front(), [&](Operation* caller) { /// TODO(aminim): this is overly conservative as some operations /// (like TPUPartitionedCallOp) may have extra operands that aren't /// propagated to the callee. return isa<CallOpInterface>(caller) && std::equal(caller->getOperandTypes().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)