Home
last modified time | relevance | path

Searched refs:GetMemorySizeInBytes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dbuffer.h47 uint64_t GetMemorySizeInBytes() const { return size_; } in GetMemorySizeInBytes() function
Dmetal_spatial_tensor.h68 uint64_t GetMemorySizeInBytes() const;
Dmetal_spatial_tensor.cc359 uint64_t MetalSpatialTensor::GetMemorySizeInBytes() const { in GetMemorySizeInBytes() function in tflite::gpu::metal::MetalSpatialTensor
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dbuffer.h49 uint64_t GetMemorySizeInBytes() const { return size_; } in GetMemorySizeInBytes() function
Dtensor.h76 uint64_t GetMemorySizeInBytes() const;
Dinference_context.cc688 total_memory += t.second.GetMemorySizeInBytes(); in GetSizeOfMemoryAllocatedForIntermediateTensors()
691 total_memory += b.GetMemorySizeInBytes(); in GetSizeOfMemoryAllocatedForIntermediateTensors()
694 total_memory += t.second.GetMemorySizeInBytes(); in GetSizeOfMemoryAllocatedForIntermediateTensors()
Dtensor.cc462 uint64_t Tensor::GetMemorySizeInBytes() const { in GetMemorySizeInBytes() function in tflite::gpu::cl::Tensor