- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 403 for _xla (0.03 sec)
-
tensorflow/compiler/jit/xla_host_send_device_context.h
#ifndef TENSORFLOW_COMPILER_JIT_XLA_HOST_SEND_DEVICE_CONTEXT_H_ #define TENSORFLOW_COMPILER_JIT_XLA_HOST_SEND_DEVICE_CONTEXT_H_ #include "xla/shape.h" #include "xla/stream_executor/device_memory.h" #include "xla/stream_executor/stream.h" #include "tensorflow/core/framework/device_base.h" #include "tfrt/concurrency/async_value_ref.h" // from @tf_runtime namespace tensorflow {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
#include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/client/local_client.h" #include "xla/executable_run_options.h" #include "xla/hlo/ir/hlo_input_output_alias_config.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/tf_pjrt_client.h" #include "xla/service/executable.h" #include "xla/service/gpu/gpu_executable_run_options.h" #include "tensorflow/core/framework/function.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_compiler_client.cc
#include "tensorflow/compiler/jit/pjrt_device_compiler_client.h" #include <memory> #include <string> #include <utility> namespace tensorflow { xla::CompileOptions GetPjRtCompileOptions( const XlaCompiler::Options& options, const XlaCompiler::CompilationResult& result) { xla::CompileOptions pjrt_compile_options; pjrt_compile_options.argument_layouts = result.xla_input_shapes; pjrt_compile_options.executable_build_options =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/BUILD
"@local_xla//xla:executable_run_options", "@local_xla//xla:status_macros", "@local_xla//xla:statusor", "@local_xla//xla/client:client_library", "@local_xla//xla/client:local_client", "@local_xla//xla/service:compiler", "@local_xla//xla/service/gpu:gpu_executable_run_options", "//tensorflow/core:core_cpu_internal", "//tensorflow/core:framework", "//tensorflow/core:lib",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
"Minimum number of operators in an XLA compilation. Ignored for " "operators placed on an XLA device or operators explicitly marked " "for compilation."), Flag("tf_xla_max_cluster_size", &mark_for_compilation_flags->tf_xla_max_cluster_size, "Maximum number of operators in an XLA compilation."), Flag( "tf_xla_ops_to_cluster",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.h
// xla_cluster_name: XLA cluster name for this XLA computation. We need it // because XLA cluster name might be different from `func_name`. // func_name_attrs: they will be used to instantiate the XLA computation func. // new_func_name: new function name for rewritten XLA computation func. // host_compute_core: mapping from outside compilation cluster name to XLA // device assignment. // fld: FunctionLibraryDefinition object.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/aot/BUILD
"@local_xla//xla:xla_data_proto_cc", "@local_xla//xla/client:client_library", "@local_xla//xla/client:compile_only_client", "@local_xla//xla/client:xla_computation", "@local_xla//xla/service:compiler", "@local_xla//xla/service/cpu:buffer_info_util", "@local_xla//xla/service/cpu:cpu_compiler", "@local_xla//xla/stream_executor:platform_manager", ], ) tf_cc_test(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 16:13:05 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/BUILD
"@local_tsl//tsl/platform:protobuf", "@local_xla//xla/client:client_library", "@local_xla//xla/client:compile_only_client", "@local_xla//xla/service/cpu:cpu_compiler", "@local_xla//xla/service/cpu:cpu_transfer_manager", "@local_xla//xla/stream_executor", "@local_xla//xla/stream_executor/host:host_platform", "@local_xla//xla/stream_executor/host:host_platform_id", ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
EXPECT_EQ(exec_args.size(), 2); std::shared_ptr<xla::Literal> literal1 = *exec_args[0]->ToLiteralSync(); EXPECT_TRUE(xla::LiteralTestUtil::Equal( *literal1, xla::LiteralUtil::CreateR2<int32_t>({{1, 2, 3}}))); std::shared_ptr<xla::Literal> literal2 = *exec_args[1]->ToLiteralSync(); EXPECT_TRUE(xla::LiteralTestUtil::Equal( *literal2, xla::LiteralUtil::CreateR2<int32_t>({{4, 5, 6}}))); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.h
#include "tensorflow/compiler/jit/variable_info.h" #include "tensorflow/compiler/jit/xla_tensor.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "xla/client/local_client.h" #include "xla/pjrt/pjrt_client.h" #include "xla/service/shaped_buffer.h" #include "xla/stream_executor/device_memory_allocator.h" #include "tensorflow/core/framework/allocation_description.pb.h" #include "tensorflow/core/framework/tensor.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 11.8K bytes - Viewed (0)