Searched refs:SetElement (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.h | 62 Value SetElement(Value index, Value buffer, Value element, OpBuilder builder,
|
| D | collection_ops_util.cc | 101 Value SetElement(Value index, Value buffer, Value element, OpBuilder builder, in SetElement() function
321 buffer = SetElement(index, buffer, slice, builder, loc); in ScatterAccumulateElements()
|
| D | tensor_list_ops_decomposition.cc | 650 cutil::SetElement(size, buffer, push.tensor(), builder, push.getLoc()); in HandleTensorListPushBackOp()
718 auto new_buffer = cutil::SetElement(index, buffer, set_item.item(), builder, in HandleTensorListSetItemOp()
|
| D | stack_ops_decomposition.cc | 432 cutil::SetElement(index, stack_val, push.elem(), builder, push.getLoc()); in HandleStackPushV2Op()
|
| D | tensor_array_ops_decomposition.cc | 278 cutil::SetElement(index_reshape, buffer, elem, builder, write.getLoc()); in HandleTensorArrayWriteV3Op()
|