- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 49 for stream_executor_ (0.23 sec)
-
tensorflow/compiler/jit/get_compiler_ir.cc
#include "xla/client/executable_build_options.h" #include "xla/client/local_client.h" #include "xla/service/hlo_graph_dumper.h" #include "xla/status_macros.h" #include "xla/stream_executor/host/host_platform_id.h" #include "xla/stream_executor/platform.h" #include "tensorflow/core/common_runtime/eager/tensor_handle.h" #include "tensorflow/core/framework/device_base.h" #include "tensorflow/core/framework/function.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_compiler_client.cc
return client_->Load(serialized_executable, build_options); } void XlaDeviceCompilerClient::WaitForProgramsToFinish() { if (client_ == nullptr) return; for (auto* executor : client_->backend().stream_executors()) { bool ok = executor->SynchronizeAllActivity(); if (!ok) { LOG(ERROR) << "Error synchronizing activity while waiting for all " "programs to complete"; } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.1K bytes - Viewed (0) -
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/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)