- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 106 for _xla (0.03 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
// eliminates the use of resource variables. // . Legalizes the operations to XLA HLO operations. // . Canonicalizes the XLA HLO operations. // // device_type: XLA JIT device to use for compilation such as "XLA_CPU_JIT", // "XLA_GPU_JIT" or "XLA_TPU_JIT". // use_tuple_args: when this is true, always create a tuple argument for the // entry computation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/BUILD
"@local_xla//xla/client:xla_computation", "@local_xla//xla/hlo/ir:hlo", "@local_xla//xla/mlir_hlo", "@local_xla//xla/mlir_hlo:hlo_dialect_registration", "@local_xla//xla/mlir_hlo:mhlo_passes", "@local_xla//xla/translate/mhlo_to_hlo:layout_util", "@local_xla//xla/translate/mhlo_to_hlo:mlir_hlo_to_hlo", "@local_xla//xla/translate/mhlo_to_hlo:type_to_shape",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
for (mlir::BlockArgument& arg : block.getArguments()) { auto num = arg.getArgNumber(); xla::Shape shape = xla::TypeToShape(arg.getType()); xla::XlaOp argop = xla::Parameter(&builder, num, shape, absl::StrCat("Arg_", num)); xla_params.push_back(argop); } std::vector<xla::XlaOp> returns(1); TF_RETURN_IF_ERROR(BuildHloFromTf(module_op, builder, xla_params, returns,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
#include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/client/compile_only_client.h" #include "xla/hlo/ir/hlo_input_output_alias_config.h" #include "xla/mlir_hlo/mhlo/IR/register.h" #include "xla/shape.h" #include "xla/shape_util.h" #include "xla/status_macros.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/op.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K 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/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/mlir/quantization/stablehlo/BUILD
"@local_xla//xla:literal", "@local_xla//xla:literal_util", "@local_xla//xla:shape_util", "@local_xla//xla/mlir_hlo", "@local_xla//xla/pjrt:pjrt_client", "@local_xla//xla/pjrt:pjrt_executable", "@local_xla//xla/pjrt/cpu:cpu_client", "@local_xla//xla/tests:literal_test_util", "@stablehlo//:chlo_ops", ], ) tf_cc_test( name = "legalize_tf_quant_test",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 28.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
#include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/stream_executor/tpu/c_api_conversions.h" #include "xla/stream_executor/tpu/status_helper.h" #include "xla/stream_executor/tpu/tpu_api.h" #include "xla/stream_executor/tpu/tpu_node_context.h" #include "xla/stream_executor/tpu/tpu_platform.h" #include "xla/stream_executor/tpu/tpu_platform_interface.h" #include "xla/stream_executor/tpu/tpu_stream_interface.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
if (xla_device == nullptr) { return errors::Internal( "Cannot get XLA metadata from non-XLA device \"", device->name(), "\". GetMetadata must only be called on an XLA device. Either an " "internal bug has been triggered, or an XLA-specific op has been " "placed on the wrong device."); } *metadata = &(xla_device->xla_metadata_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
#include "tensorflow/compiler/tf2xla/xla_helpers.h" #include "xla/util.h" #include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/dma_helper.h" #include "tensorflow/core/framework/tensor_reference.h" #include "tsl/platform/statusor.h" namespace tensorflow { // The allocator used for Tensors assigned to the XLA device. XlaDeviceAllocator::XlaDeviceAllocator(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0)