Searched refs:XlaScatter (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.h | 46 xla::StatusOr<xla::XlaOp> XlaScatter(
|
| D | scatter.cc | 33 xla::StatusOr<xla::XlaOp> XlaScatter( in XlaScatter() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 116 XlaScatter(buffer, updates, indices, in Compile()
166 auto result = XlaScatter(buffer, updates, indices, in CompileTensorScatter()
|
| D | sparse_to_dense_op.cc | 76 auto result = XlaScatter(buffer, sparse_values, indices, in Compile()
|
| D | transpose_op.cc | 185 auto result = XlaScatter(iota, iota, indices, in InvertPermutation()
|
| D | segment_reduction_ops.cc | 118 auto result = XlaScatter(buffer, /*updates=*/data, indices, in Compile()
|
| D | variable_ops.cc | 170 auto result = XlaScatter(var_value, updates, indices, indices_are_vectors_, in Compile()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 109 auto result = XlaScatter(buffer, /*updates=*/data, indices, in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17977 def TF_XlaScatterOp : TF_Op<"XlaScatter", [NoSideEffect]> {
|