- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 309 for _xla (0.13 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.cc
return absl::OkStatus(); } TF_ASSIGN_OR_RETURN( auto hlo_module_config, xla::HloModule::CreateModuleConfigFromProto( compilation_result->computation->proto(), xla::DebugOptions())); TF_ASSIGN_OR_RETURN( std::unique_ptr<xla::HloModule> hlo_module, xla::HloModule::CreateFromProto(compilation_result->computation->proto(), hlo_module_config));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 00:40:46 UTC 2024 - 6.8K 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/mlir/tf2xla/internal/legalize_tf_mlir.cc
#include "tensorflow/compiler/tf2xla/layout_util.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/mlir_hlo/mhlo/IR/register.h" #include "xla/shape.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/platform/errors.h" #include "tensorflow/core/platform/profile_utils/cpu_utils.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_compiler_client.cc
#include <memory> #include <string> #include <utility> #include <vector> #include "xla/client/local_client.h" namespace tensorflow { namespace { std::vector<const xla::Shape*> GetShapePointers( absl::Span<const xla::Shape> shapes) { std::vector<const xla::Shape*> shape_ptrs; shape_ptrs.reserve(shapes.size()); for (const auto& shape : shapes) { shape_ptrs.push_back(&shape);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
# Check the converted model in the XLA opset. quantization_options = quant_opts_pb2.QuantizationOptions( quantization_method=quant_opts_pb2.QuantizationMethod( preset_method=_PresetMethod.METHOD_STATIC_RANGE_INT8 ), tags=tags, signature_keys=[signature_key], op_set=quant_opts_pb2.XLA, enable_two_input_tensors=not use_kernel, )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_passes.td
"quant::QuantizationDialect", "shape::ShapeDialect", "sparse_tensor::SparseTensorDialect", "stablehlo::StablehloDialect" ]; } def LegalizeTFCollective : Pass<"xla-legalize-tf-collective", "ModuleOp"> { let summary = "Legalize TF/XLA collective ops (TensorFlow dialect) to the HLO dialect"; let constructor = "mlir::mhlo::CreateLegalizeTFCollectivePass()";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 17:44:14 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
#include "tensorflow/compiler/mlir/tf2xla/transforms/passes.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/mlir_hlo/mhlo/transforms/rewriters.h" #include "xla/mlir_hlo/mhlo/utils/type_conversion.h" namespace mlir { namespace odml { class TFToMhloPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/BUILD
"@local_tsl//tsl/platform:status", "@local_xla//xla/stream_executor", "@local_xla//xla/stream_executor:executor_cache", "@local_xla//xla/stream_executor:host_memory_allocation", "@local_xla//xla/stream_executor:memory_allocation", "@local_xla//xla/stream_executor:platform", "@local_xla//xla/stream_executor:stream_executor_common", ], ) cc_library(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 00:27:07 UTC 2024 - 3.1K 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)