- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 49 for tensorlist (0.57 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/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/transforms/lower_static_tensor_list.cc
// Returns a `llvm::DenseMap` which maps from the index of tensorlist in the // result, to the index of the same tensorlist in the arguments. For `If` op's // branch functions, the results and arguments are not usually matched 1-1. This // will let us konw which tensorlist result maps to which tensorlist in the // arguments. Once we know this info it will help us decide the types of the // result tensorlist based on the operand's of the `If` op.
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/lite/transforms/legalize_tensorlist.cc
}); return has_variant_input || has_variant_output; } // There are 2 standard tf ops which are not TensorList ops that may take as // input a tensorlist. These are tf.AddN and tf.ZeroesLike. Since the runtime // implementation of a tensorlist are not compatible between tf and tflite // we cannot use tflite tensorlist kernels until these cases are handled. bool IsNonTensorListVariantOp(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
if (a.TypeId() == tensorflow::TypeIndex::Make<::tensorflow::TensorList>()) { TF_RETURN_IF_ERROR(ValidateVariantType<::tensorflow::TensorList>(a)); *out = ::tensorflow::TensorList(); return ::tensorflow::TensorListBinaryAdd( cc_ctx, *a.get<::tensorflow::TensorList>(), *b.get<::tensorflow::TensorList>(), out->get<::tensorflow::TensorList>(), cc_binary_add);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.td
"legalized by this pass, then the IR won't be changed so that " "tensorlist ops can pass through (default false).">, Option< "default_to_single_batch_", "default-to-single-batch", "bool", "false", "When specified to true, if the tensorlist ops has unspecified batch " "size, this pass will assume that the batch size is one to proceed "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 22.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
OpBuilder builder(list); llvm::SmallVector<int64_t, 8> element_shape; // Infer TensorList element shape from the return type first, and then from // the const element shape operand. We first check the return type because // shape inference might have successfully inferred the element shape from // write operations on the TensorList. if (failed(GetElementShapeFromResultType(list.getType(), &element_shape))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K 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/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/tensorflow/transforms/shape_inference.cc
} }; while (!worklist.empty()) { tensorlist = worklist.top(); worklist.pop(); // TensorLists are semantically immutable. For example, TensorListSetItem // takes a TensorList as input and produces a TensorList as output. So to // traverse modifications to TensorList and verify that all elements written
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)