Home
last modified time | relevance | path

Searched refs:MemoryForAllocation (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc94 static StatusOr<MaybeOwningDeviceMemory> MemoryForAllocation( in MemoryForAllocation() function
141 buffers[i], MemoryForAllocation(allocation, arguments, memory_allocator, in CreateBufferTable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.cc1227 static StatusOr<std::shared_ptr<MaybeOwningCpuMemory>> MemoryForAllocation( in MemoryForAllocation() function
1265 TF_ASSIGN_OR_RETURN(buffers[i], MemoryForAllocation(allocation, arguments)); in CreateBufferTable()