Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for XlaDynamicUpdateSlice (0.41 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/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir

        // CHECK: %[[DIM1:.*]] = mhlo.reshape %[[SLICE1]] : (tensor<1xi32>) -> tensor<i32>
    
        // CHECK: mhlo.dynamic_update_slice %[[ARG0]], %[[ARG1]], %[[DIM0]], %[[DIM1]]
    
        %0 = "tf.XlaDynamicUpdateSlice"(%arg0, %arg1, %arg2) : (tensor<3x4xi32>, tensor<2x2xi32>, tensor<2xi32>) -> tensor<3x4xi32>
        func.return %0: tensor<3x4xi32>
      }
    
      // CHECK-LABEL: @sparse_to_dense
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

      // |index| is expanded to [index, 0, 0, ...] as the indices to input_handle.
      // On a high level, it's doing something like:
      // def : Pat<(TensorListSetItem($input_handle, $index, $item)),
      //           (XlaDynamicUpdateSlice($input_handle, ExpandDims($item, 0),
      //              Concat(ExpandDims($index, 0), [0, 0, 0, ...])))>
      LogicalResult matchAndRewriteImplWithDynamicUpdateSlice(
          TF::TensorListSetItemOp op, OpAdaptor adaptor,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

    // CHECK:         %0 = "tf.Pack"(%arg2, %arg3, %arg4) <{axis = 0 : i64}> : (tensor<i32>, tensor<i32>, tensor<i32>) -> tensor<3xi32>
    // CHECK:         %1 = "tf.XlaDynamicUpdateSlice"(%arg0, %arg1, %0) : (tensor<28x1x100xf32>, tensor<1x1x100xf32>, tensor<3xi32>) -> tensor<28x1x100xf32>
    // CHECK:         return %1 : tensor<28x1x100xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK: [[DUS:%.+]] = mhlo.dynamic_update_slice %arg0, %arg1, [[RESHAPE0]], [[RESHAPE1]] : (tensor<4x16xf32>, tensor<2x4xf32>, tensor<i32>, tensor<i32>) -> tensor<4x16xf32>
      // CHECK: return [[DUS]]
      %0 = "tf.XlaDynamicUpdateSlice"(%arg0, %arg1, %arg2) : (tensor<4x16xf32>, tensor<2x4xf32>, tensor<2xi32>) -> tensor<4x16xf32>
      func.return %0 : tensor<4x16xf32>
    }
    
    // -----
    
    // CHECK-LABEL: xla_dynamic_update_slice2
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

              op.getLoc(), op.getType(), input, std::get<1>(pair), input_indices);
        }
    
        rewriter.replaceOp(op, input);
        return success();
      }
    };
    
    // Converts a TF XlaDynamicUpdateSlice op to DynamicUpdateSlice HLO.
    class ConvertXlaDynamicUpdateSliceOp
        : public OpRewritePattern<TF::XlaDynamicUpdateSliceOp> {
     public:
      using OpRewritePattern::OpRewritePattern;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    }
    
    def TF_XlaDynamicUpdateSliceOp : TF_Op<"XlaDynamicUpdateSlice", [Pure, TF_NoConstantFold]> {
      let summary = "Wraps the XLA DynamicUpdateSlice operator, documented at";
    
      let description = [{
    https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice
    .
    
    XlaDynamicUpdateSlice generates a result which is the value of the `input`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top