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

  /external/tensorflow/tensorflow/core/kernels/
tensor_array.cc 56 #define TENSOR_ARRAY_SET_ZERO(Device, T) \
64 #define TENSOR_ARRAY_SET_ZERO_CPU(T) TENSOR_ARRAY_SET_ZERO(CPUDevice, T)
70 #define TENSOR_ARRAY_SET_ZERO_GPU(T) TENSOR_ARRAY_SET_ZERO(GPUDevice, T)
78 #undef TENSOR_ARRAY_SET_ZERO
tensor_array.h 79 #define TENSOR_ARRAY_SET_ZERO(Device, T) \
83 #define TENSOR_ARRAY_SET_ZERO_CPU(T) TENSOR_ARRAY_SET_ZERO(CPUDevice, T)
89 #define TENSOR_ARRAY_SET_ZERO_GPU(T) TENSOR_ARRAY_SET_ZERO(GPUDevice, T)
97 #undef TENSOR_ARRAY_SET_ZERO

Completed in 70 milliseconds