- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for BEF (0.13 sec)
-
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/BUILD
# ], # ) # copybara:uncomment_end tf_cc_shared_test( name = "kernel_fallback_compat_test", srcs = ["kernel_fallback_compat_test.cc"], data = [ "create_op.mlir.bef", "custom_thread_pool.mlir.bef", ], tags = ["no_oss"], deps = [ "//tensorflow/compiler/mlir/tfrt/ir:tfrt_fallback_async_opdefs", "//tensorflow/core:all_kernels", "//tensorflow/core:lib",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batch_function_fallback_benchmark_test.cc
host->Await(results); for (auto& result : results) { EXPECT_FALSE(result->IsError()); } } // Runs a BEF function that batches a function that does nothing just to measure // the runtime overhead. The BEF function signature is adapted from a real model // and is useful for benchmarking ops with large attributes and many // input/output.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 08 08:08:48 UTC 2023 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.cc
CHECK(mlir::succeeded(pm.run(*module))) << "Failed to lower module to TFRT: " << os.str(); // Convert module to BEF. bef_buffer_ = tfrt::ConvertMLIRToBEF(*module, /*disable_optional_sections=*/false); CHECK(!bef_buffer_.empty()) << "Failed to convert module to BEF"; bef_file_ = BEFFile::Open(bef_buffer_, host_context_->GetKernelRegistry(),
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/function/function.cc
#include "tensorflow/compiler/mlir/tfrt/transforms/tfrt_pipeline_options.h" #include "tensorflow/core/platform/errors.h" #include "tensorflow/core/platform/status.h" #include "tfrt/bef/bef_buffer.h" // from @tf_runtime #include "tfrt/bef_converter/mlir_to_bef.h" // from @tf_runtime namespace tensorflow { Status CompileTFMLIRToBEF(const TfrtFunctionCompileOptions& options,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 08:13:15 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/function/function.h
#include "absl/strings/string_view.h" #include "mlir/IR/BuiltinTypes.h" // from @llvm-project #include "tensorflow/compiler/mlir/tfrt/translate/tfrt_compile_options.h" #include "tensorflow/core/platform/status.h" #include "tfrt/bef/bef_buffer.h" // from @tf_runtime #include "tfrt/core_runtime/tensor_handle.h" // from @tf_runtime namespace tfrt { class CoreRuntime; } namespace mlir { class ModuleOp; } namespace tensorflow {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 04:50:20 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.td
}]; let arguments = (ins TFRT_ChainType:$in_ch, I64Attr:$num_args, StrAttr:$device, ArrayAttr:$op_attrs, // TODO(b/173025975): consider using DictionaryAttr after we support // BEF conversion for this type. ArrayAttr:$op_func_attrs, I64Attr:$op_key, StrAttr:$op_name ); let results = (outs TFRT_ChainType:$out_ch ); let hasVerifier = 1;
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/tests/tfrt_fallback/async_op_thread.mlir
// RUN: tfrt_fallback_translate -mlir-to-bef %s | tf_bef_executor --work_queue_type=mstd 2>&1 | FileCheck %s func.func @test_async_op_kernel_thread() -> !tfrt.chain { %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")
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/BUILD
"@llvm-project//llvm:Support", "@llvm-project//mlir:Parser", "@llvm-project//mlir:Pass", "@tf_runtime//:basic_kernels_alwayslink", "@tf_runtime//:bef", "@tf_runtime//:befexecutor", "@tf_runtime//:core_runtime_alwayslink", "@tf_runtime//:hostcontext", "@tf_runtime//:mlirtobef", "@tf_runtime//:support", ], )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 19:04:21 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batching_fallback.mlir
// RUN: tfrt_fallback_translate --mlir-to-bef %s | tf_bef_executor --work_queue_type=mstd:1,1 | FileCheck %s // RUN: tfrt_fallback_translate --mlir-to-bef %s | tf_bef_executor --work_queue_type=mstd:8 | FileCheck %s func.func @matmul_cpu(%ch: !tfrt.chain, %a: !tfrt_fallback.tf_tensor, %b: !tfrt_fallback.tf_tensor) -> (!tfrt.chain, !tfrt_fallback.tf_tensor) { // Enqueue a sleep onto blocking work queue, %ch0 is fulfilled when sleeping is done. %us = tfrt.constant.i32 1000
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/runtime_fallback/runtime_fallback_executor.h
#include "llvm/ADT/ArrayRef.h" #include "llvm/ADT/SmallVector.h" #include "llvm/ADT/StringRef.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/platform/threadpool_interface.h" #include "tfrt/bef/bef_buffer.h" // from @tf_runtime #include "tfrt/bef_executor/bef_file.h" // from @tf_runtime #include "tfrt/host_context/execution_context.h" // from @tf_runtime #include "tfrt/host_context/host_context.h" // from @tf_runtime
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 14 22:40:40 UTC 2022 - 2.4K bytes - Viewed (0)