- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TPUPartitionedCallOp (0.14 sec)
-
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/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)