Home
last modified time | relevance | path

Searched refs:tensor_scatter_update (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dscatter_nd_ops_test.py787 assigned = array_ops.tensor_scatter_update(t, indices, updates)
805 lambda x: array_ops.tensor_scatter_update(x, indices, updates), [x])
815 lambda updates: array_ops.tensor_scatter_update(x, indices, updates),
833 assigned = array_ops.tensor_scatter_update(t, indices, updates)
892 return array_ops.tensor_scatter_update(t, indices, updates)
905 updated = array_ops.tensor_scatter_update(tensor, indices, updates)
918 b = array_ops.tensor_scatter_update(a, [[5], [5]], [[4], [8]])
929 b = array_ops.tensor_scatter_update(
/external/tensorflow/tensorflow/python/autograph/operators/
Dslices.py140 return gen_array_ops.tensor_scatter_update(target, ((i,),), (x,))
/external/tensorflow/tensorflow/python/ops/
Dsort_ops.py229 transposition = array_ops.tensor_scatter_update(
Darray_grad.py1123 tensor_grad = array_ops.tensor_scatter_update(
Darray_ops.py5728 return gen_array_ops.tensor_scatter_update(
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py823 perm = array_ops.tensor_scatter_update(perm, [[axis1], [axis2]],
882 perm = array_ops.tensor_scatter_update(
1290 perm = array_ops.tensor_scatter_update(perm, [[ax1], [ax2]], [ax2, ax1])
1414 arr_shape = array_ops.tensor_scatter_update(arr_shape_original, [[axis]], [1])
1415 indices_shape = array_ops.tensor_scatter_update(indices_shape_original,
1419 arr_shape = array_ops.tensor_scatter_update(broadcasted_shape, [[axis]],
1421 indices_shape = array_ops.tensor_scatter_update(
1722 update_op = array_ops.tensor_scatter_update
/external/tensorflow/tensorflow/compiler/tests/
Dscatter_nd_op_test.py221 self._runScatter(array_ops.tensor_scatter_update),
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc805 Value tensor_scatter_update = builder.create<TF::TensorScatterUpdateOp>( in HandleTensorListScatterIntoExistingListOp() local
807 scatter.output_handle().replaceAllUsesWith(tensor_scatter_update); in HandleTensorListScatterIntoExistingListOp()
810 (*buffer_to_size)[tensor_scatter_update] = size; in HandleTensorListScatterIntoExistingListOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf-ops.mlir3489 func @tensor_scatter_update(%tensor: tensor<f32>, %indices: tensor<4x2xi32>, %updates: tensor<4x4xf…
3497 func @tensor_scatter_update(%tensor: tensor<4x4x4xf32>, %indices: tensor<i32>, %updates: tensor<4x4…
3505 func @tensor_scatter_update(%tensor: tensor<4x4x4xf32>, %indices: tensor<4x2xi32>, %updates: tensor…
3513 func @tensor_scatter_update(%tensor: tensor<4xf32>, %indices: tensor<4x2xi32>, %updates: tensor<4x4…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2376 name: "tensor_scatter_update"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir4293 // CHECK-LABEL: @tensor_scatter_update
4294 func @tensor_scatter_update(%tensor: tensor<?x?x?xf32>, %indices: tensor<?x2xi32>, %updates: tensor…