Searched refs:XlaScatter (Results 1 – 7 of 7) 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 | sparse_to_dense_op.cc | 76 auto result = XlaScatter(buffer, sparse_values, indices, in Compile()
|
| D | scatter_nd_op.cc | 112 XlaScatter(buffer, updates, indices, in Compile()
|
| D | transpose_op.cc | 161 auto result = XlaScatter(iota, iota, indices, in Compile()
|
| D | segment_reduction_ops.cc | 86 auto result = XlaScatter(buffer, /*updates=*/data, indices, in Compile()
|
| D | variable_ops.cc | 172 auto result = XlaScatter(var_value, updates, indices, indices_are_vectors_, in Compile()
|