- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 47 for stream_executor_ (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/BUILD
"@llvm-project//mlir:IR", "@llvm-project//mlir:Support", "@local_xla//xla:shape_util", "@local_xla//xla/mlir_hlo", "@local_xla//xla/stream_executor/tpu:c_api_conversions", "@local_xla//xla/stream_executor/tpu:tpu_api", "@local_xla//xla/translate/mhlo_to_hlo:type_to_shape", ], ) cc_library( name = "tf_saved_model_asset_sinking_pass",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 35.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cpu_device.cc
#include "tensorflow/compiler/jit/xla_device.h" #include "tensorflow/compiler/jit/xla_device_ops.h" #include "tensorflow/compiler/tf2xla/layout_util.h" #include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/stream_executor/platform_manager.h" #include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/lib/core/status.h" namespace tensorflow { using tensorflow::IdentityShapeRepresentationFn;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/BUILD
[], otherwise = [ "@local_xla//xla/stream_executor/cuda:all_runtime", "@local_xla//xla/stream_executor/cuda:cuda_driver", "@local_xla//xla/stream_executor/cuda:cuda_platform", "@local_xla//xla/stream_executor/cuda:cudnn_plugin", "@local_xla//xla/stream_executor/cuda:cufft_plugin", "@local_xla//xla/stream_executor:cuda_platform",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 16:51:59 UTC 2024 - 53.5K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.h
#include "tensorflow/compiler/jit/device_compiler.h" #include "tensorflow/compiler/jit/xla_device.h" #include "tensorflow/compiler/jit/xla_launch_util.h" #include "tensorflow/compiler/jit/xla_platform_info.h" #include "xla/stream_executor/integrations/tf_allocator_adapter.h" #include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/tensor.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 23:44:26 UTC 2023 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.h
#include <optional> #include <string> #include "tensorflow/compiler/jit/device_compiler.h" #include "tensorflow/compiler/jit/pjrt_base_device.h" #include "tensorflow/compiler/jit/xla_device.h" #include "xla/stream_executor/integrations/tf_allocator_adapter.h" #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { // Holds some information about the platform on which an
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/BUILD
"@llvm-project//mlir:Pass", "@local_tsl//tsl/platform:statusor", "@local_xla//xla:shape_util", "@local_xla//xla/client:client_library", "@local_xla//xla/stream_executor", ], ) cc_library( name = "clustering_bridge_passes", srcs = ["clustering_bridge_passes.cc"], hdrs = ["clustering_bridge_passes.h"], deps = [
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 23:42:09 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
if (it != state.allocators_.end()) { return it->second.get(); } std::unique_ptr<XlaDeviceAllocator> alloc = std::make_unique<XlaDeviceAllocator>( backend->stream_executors()[device_ordinal]); XlaDeviceAllocator* alloc_ptr = alloc.get(); state.allocators_[{backend, device_ordinal}] = std::move(alloc); return alloc_ptr; } namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/aot/compile.cc
#include "xla/client/client_library.h" #include "xla/client/compile_only_client.h" #include "xla/client/xla_computation.h" #include "xla/service/cpu/cpu_compiler.h" #include "xla/statusor.h" #include "xla/stream_executor/platform_manager.h" #include "xla/util.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/lib/core/errors.h" #include "tensorflow/core/lib/io/path.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 08:28:57 UTC 2024 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc
#include "tensorflow/compiler/tf2xla/layout_util.h" #include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/client/client_library.h" #include "xla/shape.h" #include "xla/stream_executor/platform_manager.h" #include "xla/translate/mhlo_to_hlo/type_to_shape.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/lib/monitoring/cell_reader.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/c/c_api_experimental_test.cc
// Load the library. TF_Status* status = TF_NewStatus(); string lib_path = tensorflow::GetDataDependencyFilepath(tensorflow::io::JoinPath( "tensorflow", "c", "experimental", "stream_executor", "test", "test_pluggable_device.so")); TF_Library* lib = TF_LoadPluggableDeviceLibrary(lib_path.c_str(), status); TF_Code code = TF_GetCode(status); string status_msg(TF_Message(status));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 17 22:27:52 UTC 2023 - 13.1K bytes - Viewed (0)