Searched defs:allocate_tensor (Results 1 – 3 of 3) sorted by relevance
39 void* allocate_tensor(const char* operation, size_t len, Allocator* allocator) { in allocate_tensor() function49 void* allocate_tensor(const char* operation, size_t len) { in allocate_tensor() function
1276 Status allocate_tensor(DataType type, const TensorShape& shape, in allocate_tensor() function
706 Status OpKernelContext::allocate_tensor( in allocate_tensor() function in tensorflow::OpKernelContext