- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 146 for _xla (0.15 sec)
-
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) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
#include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/client/xla_builder.h" #include "xla/client/xla_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/service/hlo.pb.h" #include "xla/translate/hlo_to_mhlo/hlo_function_importer.h" #include "xla/translate/hlo_to_mhlo/hlo_to_mlir_hlo.h" #include "xla/translate/mhlo_to_hlo/type_to_shape.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/aot/compile.cc
#include "tensorflow/compiler/tf2xla/tf2xla_util.h" #include "xla/client/client_library.h" #include "xla/client/compile_only_client.h" #include "xla/client/xla_computation.h" #include "xla/service/cpu/cpu_compiler.h" #include "xla/statusor.h" #include "xla/stream_executor/platform_manager.h" #include "xla/util.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/graph.pb.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 08:28:57 UTC 2024 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
} return mlir::success(); } bool UnsupportedPartitionedShardingType(xla::OpSharding::Type sharding) { return sharding != xla::OpSharding::REPLICATED && sharding != xla::OpSharding::OTHER; } } // namespace absl::StatusOr<std::map<int, int>> GetDimensionIndicesAndNumSplitsFromSharding( const xla::OpSharding& sharding) { int64_t tensor_tile_rank = sharding.tile_assignment_dimensions_size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h" #include "tensorflow/compiler/mlir/tensorflow/utils/convert_type.h" #include "tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h" #include "xla/xla.pb.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/tensor_shape.pb.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/core/status.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
#include "tensorflow/compiler/mlir/tf2xla/transforms/test_utils.h" #include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/client/xla_builder.h" #include "xla/client/xla_computation.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/shape_util.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/op_kernel.h" #include "tsl/lib/core/status_test_util.h" #include "tsl/platform/errors.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
// This is the argument used to refer to the pass in // the textual format (on the commandline for example). return "quant-convert-tf-xla-op-to-tf-op"; } StringRef getDescription() const final { // This is a brief description of the pass. return "Apply converting Tensorflow Xla ops to non-xla ops."; } void getDependentDialects(DialectRegistry& registry) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen_test_h.golden
::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{386ULL, 1U, ~0U}), ::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{5ULL, ~0U, ~0U}), ::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{386ULL, 2U, ~0U}), ::xla::cpu_function_runtime::BufferInfo(::xla::cpu_function_runtime::EncodedBufferInfo{5ULL, ~0U, ~0U}),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen.cc
#include "tensorflow/compiler/tf2xla/tf2xla_util.h" #include "xla/cpu_function_runtime.h" #include "xla/service/compiler.h" #include "xla/service/cpu/buffer_info_util.h" #include "xla/shape_util.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/lib/core/errors.h" namespace tensorflow { namespace tfcompile { namespace { using BufferInfo = xla::cpu_function_runtime::BufferInfo; bool IsAlpha(char c) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K bytes - Viewed (0)