- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 49 for stream_executor_ (0.24 sec)
-
tensorflow/c/kernels.h
// OpKernelContext routines // TF_GetStream returns the SP_Stream available in ctx. // This function returns a stream only for devices registered using the // StreamExecutor C API // (tensorflow/c/experimental/stream_executor/stream_executor.h). It will return // nullptr and set error status in all other cases. // Experimental: this function doesn't have compatibility guarantees and subject // to change at any time.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K 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_platform_info.cc
#include "tensorflow/compiler/jit/xla_device_compiler_client.h" #include "xla/client/client_library.h" #include "xla/client/local_client.h" #include "xla/pjrt/pjrt_client.h" #include "xla/service/compiler.h" #include "xla/stream_executor/platform_manager.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/resource_mgr.h" #include "tensorflow/core/framework/types.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K 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/c/kernels.cc
#include "absl/status/status.h" #include "absl/strings/str_cat.h" #include "tensorflow/c/c_api.h" #include "tensorflow/c/c_api_internal.h" #include "tensorflow/c/c_api_macros.h" #include "tensorflow/c/experimental/stream_executor/stream_executor.h" #include "tensorflow/c/tf_buffer.h" #include "tensorflow/c/tf_buffer_internal.h" #include "tensorflow/c/tf_datatype.h" #include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status_helper.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/BUILD
"@local_xla//xla/mlir_hlo:convert_op_folder", "@local_xla//xla/mlir_hlo:mhlo_passes", "@local_xla//xla/mlir_hlo:type_conversion", "@local_xla//xla/stream_executor/tpu:c_api_conversions", "@local_xla//xla/stream_executor/tpu:tpu_api", "@local_xla//xla/translate/hlo_to_mhlo:attribute_importer", "@local_xla//xla/translate/mhlo_to_hlo:type_to_shape", "@stablehlo//:chlo_ops",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 17.4K bytes - Viewed (0) -
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)