- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 22 for stream_executor_ (0.16 sec)
-
tensorflow/compiler/jit/xla_host_recv_device_context.h
#ifndef TENSORFLOW_COMPILER_JIT_XLA_HOST_RECV_DEVICE_CONTEXT_H_ #define TENSORFLOW_COMPILER_JIT_XLA_HOST_RECV_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.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decode_attributes_hook.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/constant_fold.h" #include "tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h" #include "xla/stream_executor/stream_executor.h" #include "tensorflow/core/framework/logging.h" #include "tsl/platform/statusor.h" namespace mlir { namespace { } // anonymous namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/infeed_ops_xla_adjust_layout.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.h" #include "xla/layout.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/shape.h" #include "xla/stream_executor/tpu/c_api_conversions.h" #include "xla/stream_executor/tpu/tpu_api.h" #include "xla/translate/mhlo_to_hlo/type_to_shape.h" namespace mlir { namespace mhlo { namespace { #define GEN_PASS_DEF_INFEEDOPSXLAADJUSTLAYOUT
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/test/BUILD
# Description: # test for stream_executor load( "//tensorflow:tensorflow.bzl", "tf_cc_shared_object", ) package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], licenses = ["notice"], ) tf_cc_shared_object( name = "test_pluggable_device.so", srcs = ["test_pluggable_device.cc"], visibility = ["//tensorflow/c:__subpackages__"], deps = [
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 17 15:20:54 UTC 2022 - 566 bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc
#include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/client/client_library.h" #include "xla/shape.h" #include "xla/stream_executor/platform.h" #include "xla/stream_executor/platform_manager.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/monitoring/cell_reader.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_gpu_device.cc
#include "tensorflow/compiler/jit/xla_platform_info.h" #include "tensorflow/compiler/tf2xla/layout_util.h" #include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/stream_executor/gpu/gpu_init.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 {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/BUILD
"//tensorflow/core:lib", "//tensorflow/core:state_ops_op_lib", "//tensorflow/core/platform:stream_executor_no_cuda", "//tensorflow/core/profiler/lib:traceme", "@local_xla//xla/stream_executor/integrations:tf_allocator_adapter", "@com_google_absl//absl/types:optional", ] # Linked by tensorflow core, without registration of jit compilation passes. cc_library(
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/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
#include "mlir/Pass/Pass.h" // from @llvm-project #include "tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.h" #include "xla/client/client_library.h" #include "xla/shape.h" #include "xla/stream_executor/platform.h" #include "xla/stream_executor/platform_manager.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/protobuf/config.pb.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K 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/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)