Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorListElementShape (0.14 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.td 

    def LegalizeTensorListElementShape : Pat<(TF_TensorListElementShapeOp $input),
    (TFL_CustomOp (Size1InputRange $input),
    (CreateStringAttr<"\"TensorListElementShape\"">), (EmptyCustomOptions))>;

def LegalizeTensorListLength : Pat<(TF_TensorListLengthOp $input),
    (TFL_CustomOp (Size1InputRange $input),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 18 07:12:51 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  3. github.com/tensorflow/tensorflow

    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)
Back to top