- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 42 for PjRt (0.04 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.h
#include <variant> #include <vector> #include "absl/status/status.h" #include "absl/types/variant.h" #include "xla/client/compile_only_client.h" #include "xla/pjrt/compile_options.pb.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/tpu/kernels/tpu_compile.pb.h" #include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h" namespace tensorflow {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
#include "tensorflow/compiler/tf2xla/shape_util.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "xla/client/local_client.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/pjrt_future.h" #include "xla/pjrt/pjrt_stream_executor_client.h" #include "xla/pjrt/tracked_device_buffer.h" #include "xla/shape_util.h" #include "xla/status_macros.h" #include "xla/stream_executor/platform_manager.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
VLOG(2) << "Compiling using PJRT"; Status status = CompileToPjRtLoadedExecutable( *ctx, platform_info_, function_, xla_compiler_args, DeviceCompileMode::kStrict, has_ref_vars_, /*may_alias_resource_update=*/true, &compilation_result, &pjrt_client, &pjrt_executable); OP_REQUIRES_OK_ASYNC(ctx, status, done); VLOG(2) << "Compiled using PJRT: " << status;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.h
#include "tensorflow/compiler/jit/xla_launch_util.h" #include "tensorflow/compiler/jit/xla_platform_info.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "xla/client/local_client.h" #include "xla/pjrt/pjrt_client.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/lib/core/status.h" namespace tensorflow {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_tensor_buffer.h
#ifndef TENSORFLOW_COMPILER_JIT_PJRT_TENSOR_BUFFER_H_ #define TENSORFLOW_COMPILER_JIT_PJRT_TENSOR_BUFFER_H_ #include <memory> #include <utility> #include "xla/pjrt/pjrt_client.h" #include "tensorflow/core/framework/allocation_description.pb.h" #include "tensorflow/core/framework/tensor.h" namespace tensorflow {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.h
#include "mlir/Pass/Pass.h" // from @llvm-project #include "tensorflow/compiler/mlir/tf2xla/api/v2/device_type.pb.h" #include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/client/compile_only_client.h" #include "xla/pjrt/compile_options.pb.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/tpu/kernels/tpu_compile.pb.h" #include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h" #include "tsl/platform/statusor.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 07:32:57 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_base_device.h
device_ordinal(device_ordinal), compilation_device_name(compilation_device_name), shape_determination_fns(shape_determination_fns) {} }; // Creates a new PJRT base device. PjRtBaseDevice(const SessionOptions& session_options, const Options& options); static absl::StatusOr<const PjRtBaseDevice::Metadata*> GetMetadataFromDevice( DeviceBase* device); private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_tensor_buffer_util_test.cc
#include <cstdint> #include <optional> #include <utility> #include <vector> #include <gtest/gtest.h> #include "tensorflow/compiler/jit/test_util.h" #include "xla/pjrt/pjrt_client.h" #include "xla/shape.h" #include "xla/shape_util.h" #include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/framework/device.h" #include "tensorflow/core/framework/tensor.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 14 18:14:47 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
#include "tensorflow/compiler/jit/variable_info.h" #include "tensorflow/compiler/jit/variable_info_util.h" #include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/tfrt_cpu_pjrt_client.h" #include "xla/tests/literal_test_util.h" #include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/framework/device.h" #include "tensorflow/core/framework/fake_input.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_tensor_buffer_util.h
#define TENSORFLOW_COMPILER_JIT_PJRT_TENSOR_BUFFER_UTIL_H_ #include <memory> #include "absl/status/statusor.h" #include "tensorflow/compiler/jit/pjrt_tensor_buffer.h" #include "xla/pjrt/pjrt_client.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor_shape.h" namespace tensorflow { // Takes the device memory pointer from the PjRtBuffer and create a Tensor that
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 14 18:14:47 UTC 2023 - 2.3K bytes - Viewed (0)