Searched refs:AddToTensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_array.h | 44 Status AddToTensor(OpKernelContext* ctx, Tensor* sum, const Tensor* current, in AddToTensor() function
53 Status AddToTensor<Device, T>(OpKernelContext * ctx, Tensor * sum, \
516 Status s = tensor_array::AddToTensor<Device, T>(ctx, existing_t, in LockedWriteOrAggregate()
523 Status s = tensor_array::AddToTensor<Device, T>(ctx, &local_tensor, in LockedWriteOrAggregate()
|
| H A D | tensor_array.cc | 33 Status AddToTensor<Device, T>(OpKernelContext * ctx, Tensor * sum, \
|