- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 382 for _xla (0.04 sec)
-
tensorflow/compiler/jit/xla_tensor.h
// REQUIRES: has_shaped_buffer() const xla::ShapedBuffer& shaped_buffer() const { CHECK(has_shaped_buffer()); return *shaped_buffer_; } xla::ShapedBuffer& shaped_buffer() { CHECK(has_shaped_buffer()); return *shaped_buffer_; } // Mutates the XlaTensor to set the ShapedBuffer. void set_shaped_buffer(xla::ScopedShapedBuffer shaped_buffer) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc
#include "tensorflow/compiler/mlir/lite/transforms/passes.h" #include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.h" #include "xla/service/hlo.pb.h" #include "xla/service/hlo_parser.h" #include "xla/translate/hlo_to_mhlo/hlo_to_mlir_hlo.h" #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/graph_debug_info.pb.h" #include "tensorflow/core/framework/types.pb.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/BUILD
"@llvm-project//mlir:IR", "@llvm-project//mlir:Support", "@local_xla//xla:shape_util", "@local_xla//xla/mlir_hlo", "@local_xla//xla/stream_executor/tpu:c_api_conversions", "@local_xla//xla/stream_executor/tpu:tpu_api", "@local_xla//xla/translate/mhlo_to_hlo:type_to_shape", ], ) cc_library( name = "tf_saved_model_asset_sinking_pass",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 35.7K bytes - Viewed (0) -
tensorflow/compiler/jit/resource_operation_safety_analysis.h
#ifndef TENSORFLOW_COMPILER_JIT_RESOURCE_OPERATION_SAFETY_ANALYSIS_H_ #define TENSORFLOW_COMPILER_JIT_RESOURCE_OPERATION_SAFETY_ANALYSIS_H_ #include "xla/service/graphcycles/graphcycles.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/graph/graph.h" namespace tensorflow { // An XLA cluster hoists all resource reads to be beginning of the cluster // execution and all the resource writes to the end. This means it cannot
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/mlir/lite/stablehlo/BUILD
"@llvm-project//mlir:TransformUtils", "@llvm-project//mlir:Transforms", "@local_xla//xla/mlir_hlo", "@local_xla//xla/mlir_hlo:hlo_dialect_registration", "@local_xla//xla/mlir_hlo:mhlo_passes", "@local_xla//xla/mlir_hlo:type_conversion", "@stablehlo//:chlo_ops", "@stablehlo//:register", ], alwayslink = 1, )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 24.9K bytes - Viewed (0) -
tensorflow/compiler/aot/compile.h
#define TENSORFLOW_COMPILER_AOT_COMPILE_H_ #include <memory> #include <string> #include "tensorflow/compiler/aot/flags.h" #include "tensorflow/compiler/tf2xla/tf2xla.pb.h" #include "xla/service/cpu/cpu_compiler.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { namespace tfcompile {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/BUILD
"@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:IR", "@local_tsl//tsl/platform:errors", "@local_xla//xla:shape_util", "@local_xla//xla/mlir_hlo:hlo_dialect_registration", "@local_xla//xla/translate/mhlo_to_hlo:type_to_shape", ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/c/BUILD
"//tensorflow/core/transforms:__subpackages__", ], deps = [ ":c_api_macros_hdrs", "@local_tsl//tsl/platform:status", "@local_xla//xla/tsl/c:tsl_status", "@local_xla//xla/tsl/c:tsl_status_internal", ] + select({ "//tensorflow:android": [ "//tensorflow/core:portable_tensorflow_lib_lite", # TODO(annarev): exclude runtime srcs ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
#include "tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "xla/client/client_library.h" #include "xla/client/compile_only_client.h" #include "xla/stream_executor/host/host_platform_id.h" #include "xla/stream_executor/platform_manager.h" #include "tensorflow/core/common_runtime/graph_constructor.h" #include "tensorflow/core/framework/function.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tensor.cc
} Status XlaTensor::AllocateShapedBuffer(DataType dtype, const xla::Shape& on_device_shape, xla::LocalClient* client, int device_ordinal) { xla::Shape on_host_shape = xla::ShapeUtil::DeviceShapeToHostShape(on_device_shape); xla::ScopedShapedBuffer shaped_buffer(on_host_shape, on_device_shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.5K bytes - Viewed (0)