- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for TensorListElementShape (0.4 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 18 07:12:51 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir
// CHECK-LABEL: listElementShape func.func @listElementShape(%arg0: tensor<!tf_type.variant<tensor<*xi32>>>) -> tensor<*xi32> { %0 = "tf.TensorListElementShape"(%arg0) : (tensor<!tf_type.variant<tensor<*xi32>>>) -> tensor<*xi32> // CHECK: %0 = "tfl.custom"(%arg0) <{custom_code = "TensorListElementShape", custom_option = #tfl<const_bytes : "0x">}> : (tensor<!tf_type.variant<tensor<*xi32>>>) -> tensor<*xi32> func.return %0 : 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/tests/constant-fold.mlir
// CHECK-LABEL: func @testTensorListElementShape func.func @testTensorListElementShape(%arg0: tensor<!tf_type.variant<tensor<2x4xf32>>>) -> tensor<2xi32> { %0 = "tf.TensorListElementShape"(%arg0) : (tensor<!tf_type.variant<tensor<2x4xf32>>>) -> tensor<2xi32> // CHECK: [[cst:%.*]] = "tf.Const{{.*}} dense<{{\[}}2, 4]> : tensor<2xi32> // CHECK-NEXT: return [[cst]] : tensor<2xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 31 23:22:24 UTC 2024 - 36.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
// CHECK: %[[SHAPE:.*]] = "tf.Const"() <{value = dense<[8, 9]> : tensor<2xi64>}> : () -> tensor<2xi64> %shape = "tf.TensorListElementShape"(%tl) : (tensor<!tf_type.variant<tensor<8x9xf32>>>) -> tensor<2xi64> // CHECK-NEXT: return %[[SHAPE]] : tensor<2xi64> func.return %shape: tensor<2xi64> } // ----- // Test tensor list gather op.
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/jit/mark_for_compilation_pass_test.cc
// because it would be trivial. Output dummy_cast = ops::Cast(root.WithOpName("test/dummy_cast"), dummy_input, DT_INT32); Output tensor_list_element_shape = ops::TensorListElementShape( root.WithOpName("test/tensor_list_element_shape"), variant_input, DT_INT32); root.graph()->AddControlEdge(dummy_cast.node(), tensor_list_element_shape.node());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
%shape_32 = "tf.TensorListElementShape"(%tl_1) : (tensor<!tf_type.variant<tensor<?x1xf32>>>) -> tensor<?xi32> %shape_64 = "tf.TensorListElementShape"(%tl_1) : (tensor<!tf_type.variant<tensor<?x1xf32>>>) -> tensor<?xi64> // CHECK: %[[CAST:.*]] = "tf.Cast"(%[[ELEMENT_SHAPE]]){{.*}}: (tensor<2xi32>) -> tensor<2xi64>
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/jit/mark_for_compilation_pass.cc
"TensorArrayScatterV3", "TensorArraySizeV3", "TensorArraySplitV3", "TensorArrayV3", "TensorArrayWriteV3", "TensorListConcatV2", "TensorListElementShape", "TensorListFromTensor", "TensorListGather", "TensorListGetItem", "TensorListLength", "TensorListPopBack", "TensorListPushBack", "TensorListReserve",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
add_to_worklist(use.getOwner()->getResult(use.getOperandNumber())); continue; } // Refining the tensor list element type might change the output of // TensorListElementShape which is expected to be the originally assigned // shape to TensorList init ops. So replace it with the original element // shape value. if (auto tl_element_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandTypeAttr shape_type = TF_DerivedOperandTypeAttr<1>; TF_DerivedResultTypeAttr element_dtype = TF_DerivedResultTypeAttr<0>; } def TF_TensorListElementShapeOp : TF_Op<"TensorListElementShape", [Pure]> { let summary = "The shape of the elements of the given list, as a tensor."; let description = [{ input_handle: the list element_shape: the shape of elements of the list }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)