Searched refs:Add5EigenImpl (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | aggregate_ops_gpu.cu.cc | 71 Add5EigenImpl<GPUDevice, T>::Compute(d, out, in1, in2, in3, in4, in5); in operator ()()
|
| H A D | aggregate_ops_cpu.h | 67 Add5EigenImpl<CPUDevice, T>::Compute(d, out, in1, in2, in3, in4, in5);
|
| H A D | aggregate_ops.h | 93 struct Add5EigenImpl { struct
|