Searched defs:allocate_tensor (Results 1 – 4 of 4) 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
59 void allocate_tensor(std::initializer_list<TensorType *> tensors) in allocate_tensor() function
1237 Status allocate_tensor(DataType type, const TensorShape& shape, in allocate_tensor() function
750 Status OpKernelContext::allocate_tensor( in allocate_tensor() function in tensorflow::OpKernelContext