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

  /external/tensorflow/tensorflow/core/kernels/
tensor_array.cc 70 #define TENSOR_ARRAY_SET_ZERO_GPU(T) TENSOR_ARRAY_SET_ZERO(GPUDevice, T)
71 TF_CALL_GPU_NUMBER_TYPES(TENSOR_ARRAY_SET_ZERO_GPU);
72 TF_CALL_complex64(TENSOR_ARRAY_SET_ZERO_GPU);
73 TF_CALL_complex128(TENSOR_ARRAY_SET_ZERO_GPU);
74 #undef TENSOR_ARRAY_SET_ZERO_GPU
tensor_array.h 89 #define TENSOR_ARRAY_SET_ZERO_GPU(T) TENSOR_ARRAY_SET_ZERO(GPUDevice, T)
90 TF_CALL_GPU_NUMBER_TYPES(TENSOR_ARRAY_SET_ZERO_GPU);
91 TF_CALL_complex64(TENSOR_ARRAY_SET_ZERO_GPU);
92 TF_CALL_complex128(TENSOR_ARRAY_SET_ZERO_GPU);
93 #undef TENSOR_ARRAY_SET_ZERO_GPU

Completed in 52 milliseconds