- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for TensorListPushBack (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
%elem = "tf._SomeOp"() : () -> tensor<f32> // CHECK-NOT: "tf.TensorListPushBack" // CHECK: %[[UPDATE:.*]] = "tf.XlaDynamicUpdateSlice" // CHECK: %[[CONST1:.*]] = "tf.Const"() <{value = dense<1> : tensor<1xi32>}> : () -> tensor<1xi32> // CHECK: %[[ADD:.*]] = "tf.AddV2"(%[[BARG2]], %[[CONST1]]) // CHECK-NOT: "tf.TensorListPushBack"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir
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/tests/shape_inference.mlir
%tl_0 = "tf.EmptyTensorList"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>> %tl_1 = "tf.TensorListPushBack"(%tl_0, %elem) : (tensor<!tf_type.variant<tensor<?x1xf32>>>, tensor<16x1xf32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>> %tl_2 = "tf.TensorListPushBack"(%tl_1, %elem) : (tensor<!tf_type.variant<tensor<?x1xf32>>>, tensor<16x1xf32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
// op into an op with a single varidic output in `.td`. auto converted = rewriter.create<TFL::CustomOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), "TensorListPushBack", TFL::ConstBytesAttr::get(getContext(), "")); rewriter.replaceOp(op, converted.getResults()); return success(); } }; struct ConvertVariantAddNOp : public OpRewritePattern<TF::AddNOp> {
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/compiler/mlir/lite/tests/lower-static-tensor-list.mlir
// CHECK: return [[RESULT]] : tensor<?x?x?xf32> } // ----- func.func @tensorlistPushBack(%arg0: tensor<3x10xf32>, %arg1: tensor<1xi32>, %arg2: tensor<10xf32>) -> tensor<?x10xf32> { %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 Oct 30 06:52:55 UTC 2023 - 39.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
auto buffer = push.getInputHandle(); auto it = buffer_to_size->find(buffer); if (it == buffer_to_size->end()) { return push.emitOpError( "found tf.TensorListPushBack on unknown TensorList."); } if (it->getSecond().fixed) { return push.emitError("cannot push on a fixed-size tensor list"); } auto size = it->getSecond().size; OpBuilder builder(push);
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/jit/encapsulate_subgraphs_pass.cc
// TODO(srbs): This filter is a little conservative. E.g. a subgraph of // the form: // Const // | // EmptyTensorList -> TensorListPushBack -> TensorListPopBack -> Op // | // (Discard popped list) //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
%max_size = "tf.Const"() {value = dense<10> : tensor<i32>} : () -> tensor<i32> %tl = "tf.EmptyTensorList"(%elem_shape, %max_size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>> %push = "tf.TensorListPushBack"(%tl, %arg0) : (tensor<!tf_type.variant<tensor<8x4xf32>>>, tensor<8x4xf32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>> return } ``` will be transformed to: ```mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
for (const mlir::OpOperand &use : arg.getUses()) { mlir::Operation *op = use.getOwner(); // Currently we only check if the tensorlist argument is consumed by // `TensorListPushBack` or `TensorListResize`, since those are the only // length-mutating ops supported in this pass. if (llvm::isa<TF::TensorListPushBackOp>(op) || llvm::isa<TF::TensorListResizeOp>(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/g3doc/_includes/tf_passes.md
%max_size = "tf.Const"() {value = dense<10> : tensor<i32>} : () -> tensor<i32> %tl = "tf.EmptyTensorList"(%elem_shape, %max_size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>> %push = "tf.TensorListPushBack"(%tl, %arg0) : (tensor<!tf_type.variant<tensor<8x4xf32>>>, tensor<8x4xf32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>> return } ``` will be transformed to: ```mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)