- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TPUPartitionedCallOp (0.28 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/mlir2graphdef/output-shapes-attr.mlir
func.func private @simple_callee(%arg0: tensor<2xi32>) -> tensor<2xi32> { %graph = tf_executor.graph { tf_executor.fetch %arg0 : tensor<2xi32> } func.return %graph#0 : tensor<2xi32> } // Test with a TPUPartitionedCallOp that is not inlined. func.func @dont_inline_tpu_partitioned_call(%arg0: tensor<2xi32> {tf._user_specified_name = "inputs_0"}, %arg1: tensor<2xi32> {tf._user_specified_name = "inputs_1"}) -> tensor<2xi32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 3.4K 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)