Searched refs:GatherElements (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.h | 96 Value GatherElements(Value indices, Value buffer, OpBuilder builder,
|
| D | collection_ops_util.cc | 266 Value GatherElements(Value indices, Value buffer, OpBuilder builder, in GatherElements() function
|
| D | tensor_array_ops_decomposition.cc | 436 cutil::GatherElements(gather.indices(), buffer, builder, gather.getLoc()); in HandleTensorArrayGatherV3Op()
|
| D | tensor_list_ops_decomposition.cc | 780 auto result = cutil::GatherElements(gather.indices(), buffer, in HandleTensorListGatherOp()
|