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