- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for PJRT (0.03 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
#include "tensorflow/compiler/tf2xla/shape_util.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "xla/client/local_client.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/pjrt_future.h" #include "xla/pjrt/pjrt_stream_executor_client.h" #include "xla/pjrt/tracked_device_buffer.h" #include "xla/shape_util.h" #include "xla/status_macros.h" #include "xla/stream_executor/platform_manager.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
VLOG(2) << "Compiling using PJRT"; Status status = CompileToPjRtLoadedExecutable( *ctx, platform_info_, function_, xla_compiler_args, DeviceCompileMode::kStrict, has_ref_vars_, /*may_alias_resource_update=*/true, &compilation_result, &pjrt_client, &pjrt_executable); OP_REQUIRES_OK_ASYNC(ctx, status, done); VLOG(2) << "Compiled using PJRT: " << status;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/BUILD
"@llvm-project//mlir:IR", ], ) cc_library( name = "tpu_passes", hdrs = ["transforms/tpu_passes.h"], visibility = [":friends"] + if_google([ "//learning/brain/tfrt/ifrt/pjrt/__subpackages__", "//learning/serving/servables/tfrt:__subpackages__", ]), deps = [ ":fallback_converter", ":tfrt_compile_options", "@llvm-project//llvm: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/BUILD
"cpu": "k8", "copt": "-mno-sse4.2", }, visibility = ["//visibility:public"], ) # This condition takes precedence over :linux_x86_64 # TODO(b/290533709): Remove this with PJRT build rule cleanup. config_setting( name = "linux_x86_64_with_weightwatcher", define_values = {"tensorflow_weightwatcher": "true"}, values = {"cpu": "k8"}, visibility = ["//visibility:public"], )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 16:51:59 UTC 2024 - 53.5K bytes - Viewed (0)