Home
last modified time | relevance | path

Searched refs:TENSOR_ARRAY_WRITE_OR_ADD_GPU (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtensor_array.cc46 #define TENSOR_ARRAY_WRITE_OR_ADD_GPU(T) TENSOR_ARRAY_WRITE_OR_ADD(GPUDevice, T) macro
47 TF_CALL_GPU_NUMBER_TYPES(TENSOR_ARRAY_WRITE_OR_ADD_GPU);
48 TF_CALL_complex64(TENSOR_ARRAY_WRITE_OR_ADD_GPU);
49 TF_CALL_complex128(TENSOR_ARRAY_WRITE_OR_ADD_GPU);
50 #undef TENSOR_ARRAY_WRITE_OR_ADD_GPU
Dtensor_array.h62 #define TENSOR_ARRAY_WRITE_OR_ADD_GPU(T) TENSOR_ARRAY_WRITE_OR_ADD(GPUDevice, T) macro
63 TF_CALL_GPU_NUMBER_TYPES(TENSOR_ARRAY_WRITE_OR_ADD_GPU);
64 TF_CALL_complex64(TENSOR_ARRAY_WRITE_OR_ADD_GPU);
65 TF_CALL_complex128(TENSOR_ARRAY_WRITE_OR_ADD_GPU);
66 #undef TENSOR_ARRAY_WRITE_OR_ADD_GPU