- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 23 for PJRT (0.04 sec)
-
tensorflow/compiler/jit/pjrt_compile_util.cc
#include "tensorflow/compiler/jit/xla_compile_util.h" #include "tensorflow/compiler/jit/xla_compiler_options_util.h" #include "tensorflow/compiler/jit/xla_platform_info.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "xla/pjrt/pjrt_client.h" #include "tensorflow/core/framework/device_base.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/resource_mgr.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_base_device.cc
options.shape_determination_fns) { if (options.shape_determination_fns.empty()) { LOG(ERROR) << "shape_representation_fns must be non-empty."; } VLOG(1) << "Created PJRT base device " << options.compilation_device_name << " device_name: " << name(); } /*static*/ absl::StatusOr<const PjRtBaseDevice::Metadata*> PjRtBaseDevice::GetMetadataFromDevice(DeviceBase* device) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_context.h
#include "tensorflow/core/framework/device_base.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { // Helper class for managing data transfers between host and accelerator // devices using PjRt. class PjRtDeviceContext : public DeviceContext { public: explicit PjRtDeviceContext( XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, bool use_pjrt_tensor_buffer = false)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 19 19:27:39 UTC 2023 - 2.7K bytes - Viewed (0) -
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_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/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)