- Sort Score
- Result 10 results
- Languages All
Results 11 - 18 of 18 for _xla (0.05 sec)
-
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
Updates inputs to TPU embedding enqueue ops depending on whether graph is in training mode or in evaluation mode. ### `-tf-tpu-validate-inputs` _Validates inputs to the TPU TF/XLA bridge_ This pass checks that the IR has valid input to TPU TF/XLA bridge. It checks the relations of multiple ops. Properties of single ops are checked by the 'verify' method of ops. ### `-tf-tpu-variable-runtime-reformatting`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
// have a valid XLA type. LogicalResult CheckClusterResults(mlir::tf_device::ClusterOp cluster) { for (OpResult result : cluster.getResults()) { if (!tensorflow::TypeValidForXLA(result.getType())) { return cluster.emitError() << "The ExtractHeadTailOutsideCompilation pass produced a Device " "cluster with a result with a non-XLA type: " << result.getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
} }; // This pass performs a manual conversion with FakeQuant, converting between // floating point and quantized space. It is designed to reproduce TF's // implementation, mirroring the previous XLA implementation. // // 1. Computing proper quantized bounds. This involves nudging the input bounds. // 2. Converting the input bounds to quantized space, rounding values. // 3. Convert back into floating point space.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
hallowlist.insert(pair.second.begin(), pair.second.end()); for (auto op : pair.second) { ASSERT_TRUE(all_ops.contains(op)); } } // Check that all registered XLA operation are in the allowlist // table or are known to not be in it. absl::flat_hash_set<string> known_not_in_list = tensorflow::testing::GetKnownXLAAllowlistOp(); std::vector<string> unknow_op;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/verification_utils.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla_passes.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/passes.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #define DEBUG_TYPE "tf-tfl-legalization" namespace mlir { namespace TFL { namespace { // Returns a TF_CastOp to I32. This function is used for CastOps that are
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis.cc
#include "absl/strings/str_join.h" #include "absl/strings/string_view.h" #include "tensorflow/compiler/jit/deadness_analysis_internal.h" #include "tensorflow/compiler/jit/xla_cluster_util.h" #include "xla/status_macros.h" #include "tensorflow/core/framework/tensor.pb.h" #include "tensorflow/core/graph/algorithm.h" #include "tensorflow/core/graph/control_flow.h" #include "tensorflow/core/graph/graph_node_util.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 60.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
op, input.getType(), scalar_zero, ArrayRef<Value>({slice1, expanded_item, slice2})); return success(); } // This function rewrites the original op into a XLA DynamicUpdateSlice op. // |item| is expanded to have the same dimension as input_handle and // |index| is expanded to [index, 0, 0, ...] as the indices to input_handle. // On a high level, it's doing something like:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/replace_cast_hacks_with_tf_xla_ops.mlir
// RUN: tf-quant-opt %s -split-input-file -inline -quant-replace-cast-hacks-with-tf-xla-ops | FileCheck %s // ----- module attributes {} { func.func @conv_with_bias_and_relu(%arg0: tensor<1x3x4x3xf32>) -> tensor<1x3x2x2xf32> { %cst = "tf.Const"() {value = dense<[162, 160]> : tensor<2xi32>} : () -> tensor<2xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 81K bytes - Viewed (0)