// CHECK: tfrt_fallback_async.executeop key({{.*}}) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.TensorArrayGatherV3"

  // CHECK: TensorArrayGatherV3
  %5 = "tf.TensorArrayGatherV3"(%handle_12, %1, %4#2) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<2x!tf_type.resource<tensor<*x!tf_type.variant>>>, tensor<i32>, tensor<f32>) -> tensor<?x!tf_type.variant>
  // CHECK: TensorArrayGatherV3

  %output_0 = "tf.TensorArrayGatherV3"(%handle_1, %indices, %res_0#3) {device = "/job:localhost/replica:0/task:0/device:CPU:0", element_shape = #tf_type.shape<?x512>} : (tensor<2x!tf_type.resource<tensor<?x512xf32>>>, tensor<?xi32>, tensor<f32>) -> tensor<?x?x512xf32>

  %indices = "tf.Const"() {value = dense<[1, 2]> : tensor<2xi32>} : () -> tensor<2xi32>
  %gather = "tf.TensorArrayGatherV3"(%ta#0, %indices, %ta#1) : (tensor<!tf_type.resource>, tensor<2xi32>, tensor<f32>) -> tensor<2x3xf32>
  func.return %gather : tensor<2x3xf32>
}

// -----

 Range 9
 RefSwitch 166
 Reshape 2
 ScatterAdd 4
 Shape 6
 ShapeN 10
 Size 2
 Snapshot 1
 StackPopV2 314
 StackPushV2 314
 StackV2 314
 StridedSlice 12
 Sub 5
 Sum 2
 Switch 247
 TensorArrayGatherV3 8
 TensorArrayGradV3 17
 TensorArrayReadV3 9
 TensorArrayScatterV3 8
 TensorArraySizeV3 4
 TensorArrayV3 8
 TensorArrayWriteV3 9
 Unique 2
 VariableV2 164
 _Retval 5
cluster 0 size 440
 Abs 40

      "StatelessTruncatedNormalV2",
      "StatelessWhile",
      "StochasticCastToInt",
      "Svd",
      "SymbolicGradient",
      "TensorArrayCloseV3",
      "TensorArrayConcatV3",
      "TensorArrayGatherV3",
      "TensorArrayGradV3",
      "TensorArrayReadV3",
      "TensorArrayScatterV3",
      "TensorArraySizeV3",
      "TensorArraySplitV3",
      "TensorArrayV3",
      "TensorArrayWriteV3",

`(n1, n2, ..., n(T-1))`.}]>:$lengths
  );

  TF_DerivedResultTypeAttr dtype = TF_DerivedResultTypeAttr<0>;
}

def TF_TensorArrayGatherV3Op : TF_Op<"TensorArrayGatherV3", []> {
  let summary = [{
Gather specific elements from the TensorArray into output `value`.
  }];

  let description = [{
All elements selected by `indices` must have the same shape.
  }];