Searched defs:AllocateTensorMemory (Results 1 – 2 of 2) sorted by relevance
34 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory() function634 absl::Status AllocateTensorMemory(const CLContext& context, const BHWC& shape, in AllocateTensorMemory() function641 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory() function
56 absl::Status AllocateTensorMemory(id<MTLDevice> device, const BHWDC& shape, in AllocateTensorMemory() function