- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TensorListPushBack (0.36 sec)
-
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/tensorflow/ir/tf_generated_ops.td
TF_VariantTensor:$output_handle, TF_Tensor:$tensor ); TF_DerivedResultTypeAttr element_dtype = TF_DerivedResultTypeAttr<1>; } def TF_TensorListPushBackOp : TF_Op<"TensorListPushBack", [Pure]> { let summary = [{ Returns a list which has the passed-in `Tensor` as last element and the other elements of the given list in `input_handle`. }]; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)