- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for XlaDynamicUpdateSlice (0.47 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tensor_array_ops_decomposition.mlir
// CHECK: %[[UPDATE1:.*]] = "tf.XlaDynamicUpdateSlice"(%[[READ1]], // CHECK: "tf.AssignVariableOp"(%[[CARG1]], %[[UPDATE1]]) // CHECK: %[[READ2:.*]] = "tf.ReadVariableOp"(%[[CARG2]]) : (tensor<!tf_type.resource<tensor<5x3xf32>>>) -> tensor<5x3xf32> // CHECK: %[[UPDATE2:.*]] = "tf.XlaDynamicUpdateSlice"(%[[READ2]], // CHECK: "tf.AssignVariableOp"(%[[CARG2]], %[[UPDATE2]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 49K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/stack_ops_decomposition.mlir
// CHECK-NEXT: %[[UPDATE_SLICE:.*]] = "tf.Reshape"(%[[PUSHVAL]], %[[UPDATE_SHAPE]]) : (tensor<f32>, tensor<1xi32>) -> tensor<1xf32> // CHECK-NEXT: %[[UPDATE:.*]] = "tf.XlaDynamicUpdateSlice"(%[[READ_VAL]], %[[UPDATE_SLICE]], %[[READ_SIZE]]) : (tensor<10xf32>, tensor<1xf32>, tensor<1xi32>) -> tensor<10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 25.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list-enable-dynamic-update-slice.mlir
// CHECK-DAG: [[EXPANDED_ITEM:%.*]] = "tf.ExpandDims"([[ITEM]], [[ZERO]]) : (tensor<10xf32>, tensor<i32>) -> tensor<*xf32> // CHECK-DAG: [[RESULT:%.*]] = "tf.XlaDynamicUpdateSlice"([[INPUT]], [[EXPANDED_ITEM]], [[ITEM_INDEX]]) : (tensor<3x10xf32>, tensor<*xf32>, tensor<?xi32>) -> tensor<3x10xf32> // CHECK-DAG: return [[RESULT]] : tensor<3x10xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 14:24:59 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
// CHECK-NEXT: %[[UPDATE_SLICE:.*]] = "tf.Reshape"(%[[PUSHVAL]], %[[UPDATE_SHAPE]]) : (tensor<f32>, tensor<1xi32>) -> tensor<1xf32> // CHECK-NEXT: %[[UPDATE:.*]] = "tf.XlaDynamicUpdateSlice"(%[[BROADCAST]], %[[UPDATE_SLICE]], %[[ZERO]]) : (tensor<10xf32>, tensor<1xf32>, tensor<1xi32>) -> tensor<10xf32> // CHECK-NEXT: %[[CONST1:.*]] = "tf.Const"() <{value = dense<1> : tensor<1xi32>}> : () -> tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_ops_on_regular_devices.cc
Name("XlaDynamicSlice").HostMemory("size_indices").Device(DEVICE), \ XlaCompileOnDemandOp); \ REGISTER_KERNEL_BUILDER(Name("XlaDynamicUpdateSlice").Device(DEVICE), \ XlaCompileOnDemandOp); \ REGISTER_KERNEL_BUILDER(Name("XlaIf").Device(DEVICE), XlaCompileOnDemandOp); \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 19 19:55:14 UTC 2022 - 8.8K bytes - Viewed (0) -
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) -
tensorflow/compiler/jit/compilability_check_util.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
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 Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0) -
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) -
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)