- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 49 for stream_executors (0.31 sec)
-
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/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/compiler/mlir/tensorflow/translate/BUILD
"@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", ], alwayslink = 1, ) cc_library( name = "split_into_island_per_op_pass",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 11K 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/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/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)