- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 194 for TFRT (0.04 sec)
-
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_sync.td
==============================================================================*/ #ifdef TFRT_FALLBACK_SYNC_OPS #else #define TFRT_FALLBACK_SYNC_OPS include "tfrt/tfrt_op_base.td" include "tfrt/core_runtime/opdefs/corert_base.td" include "tfrt/core_runtime/opdefs/corert_traits.td" include "tfrt_fallback.td" include "mlir/Interfaces/SideEffectInterfaces.td" include "tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 07 21:12:01 UTC 2023 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/fallback.mlir
// RUN: tf-tfrt-opt -tf-to-tfrt %s | FileCheck %s --dump-input=fail --dump-input-filter=all // RUN: tf-tfrt-opt -pass-pipeline='builtin.module(tf-to-tfrt{target-tpurt=true tpu-use-core-selector=false})' %s | FileCheck %s --dump-input=fail --dump-input-filter=all // CHECK-LABEL: func @_tfrt_fallback_init // CHECK-SAME: {{.*}} !tfrt.chain // CHECK: tfrt_fallback_async.createop(%arg0) key(0) device("/device:CPU:0") "tf.ParseExampleV2"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/runtime_lowering_gpu.mlir
// RUN: tf-tfrt-opt -tfrt-lower-cluster-to-runtime-ops-non-tpu -split-input-file -verify-diagnostics %s | FileCheck %s module attributes {tf.versions = {producer = 888 : i32}, tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:TPU_SYSTEM:0", "/job:worker/replica:0/task:0/device:GPU:0"]} { // CHECK-LABEL: @converts_cluster func.func @converts_cluster() { // CHECK: "tf.XlaLaunch"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 17:41:44 UTC 2023 - 840 bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/BUILD
cc_api_version = 2, visibility = [ "//learning/serving/contrib/tfrt/mlir/saved_model_analysis:__pkg__", ], ) cc_library( name = "cluster_tf", srcs = ["cluster_tf.cc"], hdrs = ["cluster_tf.h"], visibility = [ "//learning/serving/contrib/tfrt/mlir/saved_model_analysis:__pkg__", "//tensorflow/compiler/mlir/tensorflow/transforms:__pkg__",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/BUILD
features = if_oss(["--path=org_tensorflow/tensorflow/compiler/mlir/tfrt"]), size_override = { "fallback.mlir": "medium", }, test_file_exts = ["mlir"], ) # Bundle together all of the test utilities that are used by tests. filegroup( name = "test_utilities", testonly = True, data = [ "//tensorflow/compiler/mlir/tfrt:tf-tfrt-opt", "@llvm-project//llvm:FileCheck",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops_mlrt.mlir
// RUN: tf-tfrt-opt -split-input-file -tfrt-lower-tf-savedmodel="hoist-invariant-ops=true fuse-get-resource-ops=false" %s | FileCheck %s --dump-input=fail --dump-input-filter=all module attributes {tf_saved_model.semantics} { // Test hoisting hash table op. // CHECK-LABEL: func @_tfrt_resource_init // CHECK: [[handle:%.*]] = "tf.HashTableV2"() // CHECK-SAME: shared_name = "x"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/testdata/test.mlir
func.func @test(%ch: !tfrt.chain, %arg0: !corert.tensorhandle, %arg1_th: !corert.tensorhandle) { %cpu = corert.get_op_handler %ch "cpu" %0 = corert.executeop(%cpu) "tf.Relu"(%arg0) { T = f32 } : 1 %arg1 = tfrt_fallback_async.corert_tensorhandle_to_fallback_tensor %arg1_th {_tfrt_cost = 1 : i64, device = "/CPU:0"} : (!corert.tensorhandle) -> (!tfrt_fallback.tf_tensor) %1 = tfrt_fallback_async.executeop key(0) cost(100) device("/CPU:0") "tf.Relu"(%arg1) { T = f32 } : 1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Dec 29 18:20:20 UTC 2022 - 496 bytes - Viewed (0) -
tensorflow/cc/saved_model/experimental/tests/saved_model_api_test.cc
return tensorflow::io::JoinPath(tensorflow::testing::TensorFlowSrcRoot(), kTestData, saved_model_dir); } // This value parameterized test allows us to test both TFRT // and non TFRT runtimes. // https://github.com/google/googletest/blob/dcc92d0ab6c4ce022162a23566d44f673251eee4/googletest/docs/advanced.md#value-parameterized-tests class CPPSavedModelAPITest : public ::testing::TestWithParam<bool> {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 14 23:59:14 UTC 2020 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/device_conversion.mlir
// RUN: tf-tfrt-opt -tf-to-tfrt %s | FileCheck %s --dump-input=fail // CHECK-LABEL: func @device_test func.func @device_test( %arg0: tensor<3x1xf32> {tf_saved_model.index_path = [0]}, %arg1: tensor<1x3xf32> {tf_saved_model.index_path = [0]}) -> (tensor<3x3xf32> {tf_saved_model.index_path = []}) { // CHECK: {{%.*}} = corert.get_op_handler %arg0 "/device:GPU:0"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 645 bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/mlrt/mlrt_ops.h
#include "mlir/IR/OpDefinition.h" // from @llvm-project #include "mlir/IR/OpImplementation.h" // from @llvm-project #include "tensorflow/compiler/mlir/tfrt/ir/mlrt/mlrt_dialect.h" #define GET_OP_CLASSES #include "tensorflow/compiler/mlir/tfrt/ir/mlrt/mlrt_ops.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 22:07:30 UTC 2023 - 1.2K bytes - Viewed (0)