HomeSort by relevance Sort by last modified time
    Searched refs:TENSOR_ARRAY_WRITE_OR_ADD_GPU (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
tensor_array.cc 46 #define TENSOR_ARRAY_WRITE_OR_ADD_GPU(T) TENSOR_ARRAY_WRITE_OR_ADD(GPUDevice, T)
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
tensor_array.h 62 #define TENSOR_ARRAY_WRITE_OR_ADD_GPU(T) TENSOR_ARRAY_WRITE_OR_ADD(GPUDevice, T)
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

Completed in 478 milliseconds