- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 232 for _xla (0.06 sec)
-
tensorflow/compiler/jit/variable_info_util.cc
// cluster because we would not handle variable updates correctly. Any // locks we have already acquired will be released when the VariableInfo // objects are destroyed. // TODO(b/128495870) Add support for passing aliased resource variables. return errors::Unimplemented("Duplicate variable passed to XLA cluster"); } if (variables[i]->read_only()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
limitations under the License. ==============================================================================*/ // This transformation pass converts stateful and stateless partitioned calls // with _xla_compile_device_type attribute to XLA launch ops. #include <stack> #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/xla_call_module_to_call.mlir
// RUN: stablehlo-quant-opt %s -split-input-file -stablehlo-xla-call-module-to-call | FileCheck %s // ----- // Tests composite tf.XlaCallModule is converted to func.call. module { // CHECK-LABEL: func.func @main func.func @main(%arg0: tensor<1x1024xf32>) -> tensor<1x3xf32> { // CHECK: call @composite_dot_general_fn_1 // CHECK-SAME: (tensor<1x1024xf32>, tensor<1024x3xf32>) -> tensor<1x3xf32> // CHECK-NOT: tf.XlaCallModule
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 20:02:00 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/convert_mhlo_quant_to_int.mlir
// RUN: tf-mlir-translate -mlir-tf-to-hlo-text %s -tf-input-shapes=: -tf-xla-emit-return-tuple | FileCheck %s module attributes {tf.versions = {producer = 179 : i32}} { func.func @main(%arg0: tensor<f32>) -> tensor<f32> { %0 = "stablehlo.uniform_quantize"(%arg0) : (tensor<f32>) -> tensor<!quant.uniform<ui8:f32, 34.0:16>> %1 = "stablehlo.uniform_dequantize"(%0) : (tensor<!quant.uniform<ui8:f32, 34.0:16>>) -> tensor<f32> func.return %1 : tensor<f32> } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 07 16:28:50 UTC 2023 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/optimize.cc
#include "stablehlo/dialect/ChloOps.h" // from @stablehlo // IWYU pragma: keep #include "tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" // IWYU pragma: keep namespace mlir::quant::stablehlo { namespace { #define GEN_PASS_DEF_OPTIMIZEINTGRAPH #include "tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 24 02:26:47 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// RUN: tf-opt %s -split-input-file -tf-xla-broadcast | FileCheck %s
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/adjust-layout.mlir
// RUN: tf-opt -pass-pipeline='builtin.module(func.func(infeed-ops-xla-adjust-layout))' %s | FILECHECK_OPTS="" FileCheck %s func.func @infeed_dequeue_tuple() -> (tensor<1x8x4x4xi32>, tensor<1x100x1xf32>) { // CHECK: [[TOKEN:%.*]] = mhlo.create_token : !mhlo.token %0 = "mhlo.create_token"() : () -> !mhlo.token // CHECK: [[INFEED:%.*]]:3 = "mhlo.infeed"([[TOKEN]]) <{ // CHECK-SAME{LITERAL}: infeed_config = "", layout = [[1, 3, 2, 0], [1, 2, 0]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 817 bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/saved_model_test.cc
&bef_buffer, model_context, fallback_state.get())); // The module contains an XLA function, as well as a while body and a while // condition within the XLA function. EXPECT_EQ(fallback_state->process_function_library_runtime() .GetFunctionLibraryDefinition() ->num_functions(), 3); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 01:17:29 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/aot/flags.cc
"fetch nodes will be dumped to stdout in a comma-separated list. " "Typically used to format arguments for other tools, e.g. " "freeze_graph."}, // Flags controlling the XLA ahead-of-time compilation, that correspond to // the fields of xla::cpu::CpuAotCompilationOptions. // // TODO(toddw): The following flags also need to be supported: // --xla_cpu_llvm_opt_level // --xla_cpu_llvm_cl_opts
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 05 16:55:24 UTC 2022 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_client_test.cc
TEST(GetExecutableOptionTest, Basic) { XlaCompiler::Options options; options.device_ordinal = 0; options.alias_passthrough_params = true; options.detailed_logging = true; XlaCompiler::CompilationResult result; xla::Shape xla_output_shape; result.xla_output_shape = xla_output_shape; auto build_option = GetExecutableBuildOptions(options, result, /*default_device_ordinal=*/-1); EXPECT_EQ(build_option.device_ordinal(), 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 20:35:26 UTC 2023 - 2.5K bytes - Viewed (0)