Searched refs:CopyTensorPluggableDevice (Results 1 – 1 of 1) sorted by relevance
96 inline void CopyTensorPluggableDevice(OpKernelContext* ctx, Tensor& src, in CopyTensorPluggableDevice() function627 ? &CopyTensorPluggableDevice<T> in Compute()796 ? &CopyTensorPluggableDevice<T> in Compute()820 const auto copy_tensor = IsPluggableDevice(c) ? &CopyTensorPluggableDevice<T> in Scatter()1103 CopyTensorPluggableDevice<T>(c, input_batch, frame); in Compute()