Searched defs:allocate_temp (Results 1 – 6 of 6) sorted by relevance
24 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void* allocate_temp(size_t num_bytes) const { in allocate_temp() function
71 EIGEN_STRONG_INLINE void* allocate_temp(size_t num_bytes) const { in allocate_temp() function
229 EIGEN_STRONG_INLINE void* allocate_temp(size_t num_bytes) const { in allocate_temp() function
174 EIGEN_STRONG_INLINE void *allocate_temp(size_t num_bytes) const { in allocate_temp() function880 EIGEN_STRONG_INLINE void *allocate_temp(size_t num_bytes) const { in allocate_temp() function
289 Status OpKernelConstruction::allocate_temp(DataType type, in allocate_temp() function in tensorflow::OpKernelConstruction308 Status OpKernelConstruction::allocate_temp(DataType type, in allocate_temp() function in tensorflow::OpKernelConstruction820 Status OpKernelContext::allocate_temp( in allocate_temp() function in tensorflow::OpKernelContext858 Status OpKernelContext::allocate_temp(DataType type, const TensorShape& shape, in allocate_temp() function in tensorflow::OpKernelContext865 Status OpKernelContext::allocate_temp(DataType type, const TensorShape& shape, in allocate_temp() function in tensorflow::OpKernelContext
153 Status TFRTOpKernelContext::allocate_temp(DataType type, in allocate_temp() function in tensorflow::TFRTOpKernelContext