Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for XlaDynamicUpdateSlice (0.26 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

    for `tf.EmptyTensorList` or the specified size for `tf.TensorListReserve`.
    Each push will be turned into `tf.XlaDynamicUpdateSlice` with the incremented
    size, and each pop will be turned into a `tf.Slice` and a copy of the buffer
    with decremented size. Each `tf.TensorListSetItem` will be turned into a
    `tf.XlaDynamicUpdateSlice` with unchanged size, and each `tf.TensorListGetItem`
    will be rewritten to a `tf.Slice`.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir

    }
    
    func.func @dynamic_update_slice(%arg0: tensor<4x5xi32>, %arg1: tensor<1x5xi32>, %arg2: tensor<2xi32>) -> tensor<4x5xi32> {
      %0 = "tf.XlaDynamicUpdateSlice"(%arg0, %arg1, %arg2) : (tensor<4x5xi32>, tensor<1x5xi32>, tensor<2xi32>) -> tensor<4x5xi32>
      func.return %0 : tensor<4x5xi32>
    
    // CHECK-LABEL:dynamic_update_slice
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 05 01:54:33 UTC 2024
    - 153.4K bytes
    - Viewed (0)
Back to top