- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for tensorlist (0.23 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir
// RUN: tf-opt -tfl-legalize-tensorlist -canonicalize -split-input-file %s | FileCheck %s // CHECK-LABEL: listReserveScalarShapeI32 func.func @listReserveScalarShapeI32(%arg0: tensor<i32>, %arg1: tensor<i32>) -> tensor<!tf_type.variant<tensor<*xi32>>> { %0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*xi32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/tensorlist_patterns.td
//===----------------------------------------------------------------------===// // TensorList transformation patterns. // Note that the pattern below rewrites `TensorList` tensors (which has type DT_VARIANT) // into regular tensors. We also assume that each element in the `TensorList` has // a same constant shape. //===----------------------------------------------------------------------===//
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 16 23:20:46 UTC 2022 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_tensor_list_init_ops_pass.cc
// Replicates the TensorList initialization ops for all the uses. // No need to delete the original TensorList as it might be used elsewhere. template <typename T> void ReplicateTensorListForUses(T tensor_list_op) { Value tensor_list = tensor_list_op.getResult(); std::vector<OpOperand*> uses; for (auto& use : tensor_list.getUses()) { uses.emplace_back(&use); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 22 17:28:34 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list-default-to-single-batch.mlir
// RUN: tf-opt "-tfl-lower-static-tensor-list=allow-tensorlist-pass-through default-to-single-batch=false" -split-input-file %s | FileCheck %s // ----- func.func @tensorlistReserveConstantUnknownElementShapeDim(%arg0: tensor<i32>, %arg1: tensor<i32>) -> tensor<?x7xf32> { %cst = arith.constant dense<[-1, 7]> : tensor<2xi32> %0 = "tf.TensorListReserve"(%cst, %arg0) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<?x7xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 14:24:59 UTC 2022 - 766 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export_flags.h
// during the flatbuffer translation. extern bool emit_builtin_tflite_ops; extern bool emit_select_tf_ops; extern bool emit_custom_ops; // The flag to control whether to lower tensorlist ops into TF ops. extern bool lower_tensor_list_ops; // The flag to control whether debug info gets stripped on export. extern bool strip_debug_info; // The flag to control whether to store constant & custom buffers inside
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 20 17:49:09 UTC 2023 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list-enable-dynamic-update-slice.mlir
// RUN: tf-opt "-tfl-lower-static-tensor-list=allow-tensorlist-pass-through default-to-single-batch enable-dynamic-update-slice" -split-input-file %s | FileCheck %s // ----- // CHECK-LABEL: tensorlistSetItem func.func @tensorlistSetItem(%arg0: tensor<3x10xf32>, %arg1: tensor<1xi32>, %arg2: tensor<i32>, %arg3: tensor<10xf32>) -> tensor<3x10xf32> { %0 = "tf.TensorListFromTensor"(%arg0, %arg1) : (tensor<3x10xf32>, tensor<1xi32>) -> tensor<!tf_type.variant<tensor<10xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 14:24:59 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/common/tfl_pass_config.h
// If `emit_builtin_tflite_ops` is true, TF Lite legalization passes will be // added, which produces TF Lite ops. bool emit_builtin_tflite_ops = true; // If `lower_tensor_list_ops` is true, tensorlist ops will be lowered to basic // TF ops before legalization to TF Lite dialect. bool lower_tensor_list_ops = false; // The allowlist of functions that would be preserved after trimming.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:05:30 UTC 2024 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc
llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> legalize_custom_tensor_list_ops( "legalize-custom-tensor-list-ops", llvm::cl::desc("Convert \"tf.TensorList*\" ops to \"tfl.custom_op\"" "if they can all be supported."), llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> serialize_stablehlo_ops( "serialize-stablehlo-ops",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_translate.cc
llvm::cl::location(emit_custom_ops), llvm::cl::init(false)); // NOLINTNEXTLINE static opt<bool, true> lower_tensor_list_ops_flag( "lower-tensor-list-ops", llvm::cl::desc("Lower the TensorList ops within the TFLite dialect"), llvm::cl::location(lower_tensor_list_ops), llvm::cl::init(false)); // NOLINTNEXTLINE static opt<bool, true> strip_debug_info_flag(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/resources/decomposition_lib.mlir
// OpDefs. tfr.func @tf__my_add_n(%values: !tfr.tensor_list, %n: i64 {tfr.name="N"}) -> !tfr.tensor { %index = arith.constant 0 : index %cst = arith.constant 1 : i64 %eq = arith.cmpi eq, %n, %cst : i64 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor %res = scf.if %eq -> !tfr.tensor { scf.yield %v1 : !tfr.tensor } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 13 16:33:28 UTC 2021 - 4.2K bytes - Viewed (0)