- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 50 for tensor_list (0.2 sec)
-
tensorflow/compiler/mlir/tfr/ir/tfr_types.h
}; class TFRTensorListType : public detail::TFRTypeImpl<TFRTensorListType> { public: using TFRBase::TFRBase; static constexpr StringLiteral name = "tfr.tensor_list"; static std::string getTypeName() { return "TFRTensorListType"; } }; class TFRAttrType : public Type::TypeBase<TFRAttrType, TFRType, TypeStorage> { public: using Base::Base;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 05 07:17:01 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
SmallVector<Value, 4> tensor_list; for (int i = res.index(); i < new_op->getNumResults(); i++) { Value new_res = new_op->getResult(i); auto casted = rewriter.create<CastOp>(loc, unconstrainted_type, new_res); tensor_list.push_back(casted.getOut()); } auto list_op = rewriter.create<BuildListOp>(loc, res_type, tensor_list); new_results.push_back(list_op.getOut());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir
// CHECK: [[tensor_list:%.*]] = "tf.TensorListReserve"([[shape:%.*]], [[reserve_size:%.*]]) { %0 = "tf.TensorListReserve"(%cst_1, %cst_2) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<i32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*xf32>>> // CHECK: [[map_fn_result:%.*]] = tf_mlrt.tf_map_fn([[reserve_size]], [[tensor_list]], %arg0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:40:22 UTC 2024 - 68.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
output << " input_index=" << xla_output_description.input_index; if (xla_output_description.is_constant) output << " constant"; if (xla_output_description.is_tensor_list) output << " tensor_list"; output << '\n'; } for (const auto& resource_update : compilation_result.resource_updates) { output << "// ResourceUpdate input_index=" << resource_update.input_index
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/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/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/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/cc/experimental/libtf/tests/tensor_test.cc
A. Unique TensorFlower <******@****.***> 1684532433 -0700
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 21:44:52 UTC 2023 - 5K bytes - Viewed (0) -
tensorflow/cc/experimental/base/tests/tensor_test.cc
A. Unique TensorFlower <******@****.***> 1712999357 -0700
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 09:56:08 UTC 2024 - 6K 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)