- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 309 for _xla (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
mlir::LogicalResult DecodeShardingAttribute(const std::string& shard_str, xla::OpSharding& sharding, bool report_error = true); // Encodes the sharding in human readable form. mlir::LogicalResult DecodeShardingAttribute(mlir::Attribute shard_attr, xla::OpSharding& sharding, bool report_error = true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/BUILD
"@local_xla//xla/client:padding", "@local_xla//xla/client:sharding_builder", "@local_xla//xla/mlir_hlo", "@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",
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/BUILD
"@local_xla//xla:status_macros", "@local_xla//xla/client:local_client", "@local_xla//xla/pjrt:pjrt_client", "@local_xla//xla/pjrt:pjrt_future", "@local_xla//xla/pjrt:pjrt_stream_executor_client", "@local_xla//xla/pjrt:tracked_device_buffer", "@local_xla//xla/service:shaped_buffer", "@local_xla//xla/stream_executor:device_memory_allocator",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 61.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
op->emitWarning("TF/XLA TPU bridge input check: found invalid op. ") << op->getName() << " can't be both xla and non-xla"; return true; } op->emitOpError("TF/XLA TPU bridge input check: found invalid op. ") << "Can't be both xla and non-xla"; return false; } return true; } void GetFlattenedShardings(llvm::SmallVector<xla::OpSharding>& shardings_result,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.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 { namespace tf2xla { namespace v1 { // Compiles the given Tensorflow graph into xla::HLO. The result is in
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/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
#include "xla/pjrt/c/pjrt_c_api.h" #include "xla/pjrt/pjrt_c_api_client.h" #include "tensorflow/core/framework/tensor.h" namespace tensorflow { absl::StatusOr<PJRT_Buffer*> GetPjRtCBufferFromTensor(const Tensor* tensor); absl::Status SetPjRtCBufferToTensor(PJRT_Buffer* c_buffer, xla::PjRtCApiClient* c_api_client,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.h
const XlaPlatformInfo& platform_info, DeviceType compilation_device_type, DeviceCompiler<xla::LocalExecutable, xla::LocalClient>** xla_device_compiler); // Fetches a DeviceCompiler from the tfrt_global resource manager (or creates // one there if not found) that uses xla::PjRtClient using an appropriate // PjRtClient for `platform_info.device_type()` and sets *pjrt_device_compiler
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/jit/ops/xla_ops.cc
.Output("results: Tresults") .Attr("Tresults: list(type) >= 0") .Attr("function: func") // XLA random-number generation ops are stateful. // TODO(phawkins): create stateful and non-stateful variants of XlaLaunch. .SetIsStateful() .Doc("XLA Launch Op. For use by the XLA JIT only."); REGISTER_OP("XlaLaunchV2") .Input("args: Targs") .Output("results: Tresults")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 09:08:06 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/jit/device_executable_persistor_test.cc
#include "tensorflow/compiler/jit/xla_compilation_cache.pb.h" #include "tensorflow/compiler/jit/xla_device_compiler_client.h" #include "xla/client/client_library.h" #include "xla/client/executable_build_options.h" #include "xla/client/local_client.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/tfrt_cpu_pjrt_client.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/errors.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_test.cc
namespace { using ::testing::_; using ::testing::Return; using XlaDeviceCompiler = DeviceCompiler<xla::LocalExecutable, xla::LocalClient>; using XlaDeviceExecutablePersistor = DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>; using Signature = DeviceCompilationClusterSignature; xla::LocalClient* GetLocalClient() { // TODO(b/255826209): Figure out how to run this test with the CPU client as // well.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 19.8K bytes - Viewed (0)