- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 422 for _xla (0.17 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
#include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/client/xla_computation.h" #include "xla/hlo/ir/hlo_sharding.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/mlir_hlo/mhlo/IR/register.h" #include "xla/mlir_hlo/mhlo/transforms/passes.h" #include "xla/shape.h" #include "xla/translate/mhlo_to_hlo/layout_util.h" #include "xla/translate/mhlo_to_hlo/mlir_hlo_to_hlo.h" #include "xla/translate/mhlo_to_hlo/type_to_shape.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
#include <vector> #include <gtest/gtest.h> #include "absl/log/check.h" #include "xla/pjrt/c/pjrt_c_api.h" #include "xla/pjrt/c/pjrt_c_api_cpu.h" #include "xla/pjrt/c/pjrt_c_api_wrapper_impl.h" #include "xla/pjrt/cpu/cpu_client.h" #include "xla/pjrt/pjrt_api.h" #include "xla/pjrt/pjrt_c_api_client.h" #include "xla/shape.h" #include "xla/shape_util.h" #include "tensorflow/core/framework/types.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 01 16:29:40 UTC 2024 - 7.2K bytes - Viewed (0) -
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)