- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TensorListElementShape (0.21 sec)
-
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/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)