Searched defs:AllocateTensorMemory (Results 1 – 2 of 2) sorted by relevance
33 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory() function650 absl::Status AllocateTensorMemory(const CLContext& context, const BHWC& shape, in AllocateTensorMemory() function657 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory() function
57 absl::Status AllocateTensorMemory(id<MTLDevice> device, const BHWDC& shape, in AllocateTensorMemory() function