- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 36 for tfrt_fallback_async (0.3 sec)
-
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.td
Example: %tensor = tfrt_fallback_async.const_string_tensor {shape = [1, 2], value = ["const", "string"]} Note: if the entries in `value` are all the same, we can just keep one copy. In other words, the following two lines are effectively the same. %tensor = tfrt_fallback_async.const_string_tensor {shape = [1, 2], value = ["string", "string"]}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 15:01:21 UTC 2024 - 15.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc
//===----------------------------------------------------------------------===// namespace { // Simplifies pattern containing a corert const tensor op followed by a // `tfrt_fallback_async.corert_tensorhandle_to_fallback_tensor` op to a single // tfrt_fallback_async const tensor. struct ConstCoreRTTensorHandleToFallbackTensorCanonicalization : public OpRewritePattern<CoreRTTensorHandleToFallbackTensorOp> { using OpRewritePattern<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.h
public: explicit FallbackAsyncDialect(MLIRContext *context); static StringRef getDialectNamespace() { return "tfrt_fallback_async"; } }; } // namespace fallback_async } // namespace tfrt #define GET_OP_CLASSES #include "tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 14:49:54 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ir/fallback_opt.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 11:03:04 UTC 2022 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/fallback_canonicalization.mlir
// CHECK: [[b:%.*]] = tfrt_fallback_async.const_dense_tensor dense<[false, true]> : tensor<2xi1> // CHECK-NEXT: [[a:%.*]] = tfrt_fallback_async.const_dense_tensor dense<[true, false]> : tensor<2xi1> // CHECK-NEXT: tfrt_fallback_async.corert_tensorhandle_to_fallback_tensor [[arg0]] // CHECK-NEXT: tfrt_fallback_async.corert_tensorhandle_to_fallback_tensor [[arg1]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 04 14:07:37 UTC 2022 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batching_fallback.mlir
%a2 = tfrt_fallback_async.const_dense_tensor dense<[[3, 3], [3, 3]]> : tensor<2x2xi32> %b = tfrt_fallback_async.const_dense_tensor dense<[[1, 1], [1, 1]]> : tensor<2x2xi32> // Two batch_size=2 batches get concatenated. %result_1 = tfrt_fallback_async.batch_function device("/device:CPU:0") @matmul_cpu (%a1, %b) { num_batch_threads = 1, max_batch_size = 4, allowed_batch_sizes = [4], batch_timeout_micros = 1000000,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 18 22:58:56 UTC 2023 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/insert_fallback_tensor_copy.mlir
// CHECK: [[value:%.*]] = tfrt_fallback_async.executeop key({{.*}}) {{.*}} "tf.AddV2"([[arg]], [[arg]]) %0 = tfrt_fallback_async.executeop key(0) cost(1024) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.AddV2"(%arg, %arg) {T = f32} : 1 // CHECK: [[copy:%.*]] = tfrt_fallback_async.copy_if_small [[value]] // CHECK: tfrt_fallback_async.executeop key({{.*}}) {{.*}} "tf.AddV2"([[copy]], [[copy]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:51:48 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/BUILD
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 27 20:43:07 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/async_op_thread.mlir
%ch0 = tfrt.new.chain %ch1 = tfrt_fallback_async.createop(%ch0) key(0) device("/CPU:0") "tf.Const"() { dtype = i32, value = dense<[2]> : tensor<1xi32> } num_args(0) %ch2 = tfrt_fallback_async.createop(%ch1) key(1) device("/CPU:0") "tf.TestAsyncTfrtAsyncThread"() {T = i32} num_args(1) %ch3 = tfrt_fallback_async.createop(%ch2) key(2) device("/CPU:0") "tf.TestPrintThreadName"() num_args(0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 18 22:58:56 UTC 2023 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/tf_to_corert_pipeline.mlir
// The predicate function should be inlined. // CHECK-DAG: tfrt_fallback_async.const_dense_tensor dense<9> : tensor<i32> // CHECK-DAG: tfrt_fallback_async.const_dense_tensor dense<0> : tensor<i32> // CHECK-NEXT: tfrt_fallback_async.executeop key({{.*}}) cost({{.*}}) device("/device:CPU:0") "tf.Less" // CHECK-NEXT: [[pred:%.*]] = tfrt_fallback_async.predicate
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 7.7K bytes - Viewed (0)