- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TensorArrayWriteV3 (0.2 sec)
-
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
// safe. if (auto tensor_array_v3_op = dyn_cast<TF::TensorArrayV3Op>(tf_op)) { // The "flow" in TensorArrayV3 is always a scalar float tensor. // https://www.tensorflow.org/api_docs/python/tf/raw_ops/TensorArrayWriteV3 Value flow = tensor_array_v3_op.getFlow(); Type scalar_f32_tensor_type = RankedTensorType::get(/*shape=*/{}, rewriter.getF32Type()); flow.setType(scalar_f32_tensor_type); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/control_flow.mlir
%y = "tf.MatMul"(%x, %cst) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<?x100xf32>, tensor<100x512xf32>) -> (tensor<?x512xf32>) %flow_1_out = "tf.TensorArrayWriteV3"(%handle_1, %index, %y, %flow_1) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<2x!tf_type.resource<tensor<?x512xf32>>>, tensor<i32>, tensor<?x512xf32>, tensor<f32>) -> tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 00:40:32 UTC 2024 - 17.5K bytes - Viewed (0)