Home
last modified time | relevance | path

Searched defs:scratch_buffer_size (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbatch_matmul.cc160 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(lhs_rank); in InitializeTemporaries() local
186 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(rhs_rank); in InitializeTemporaries() local
H A Dunidirectional_sequence_lstm.cc972 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
1201 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
H A Dlstm.cc1419 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
1776 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
1814 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
/aosp_15_r20/external/mesa3d/src/imagination/vulkan/
H A Dpvr_device.c2886 uint64_t scratch_buffer_size; in pvr_CreateFramebuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc418 size_t scratch_buffer_size = Eigen::kGpuScratchSize + sizeof(unsigned int); in InitScratchBuffers() local
/aosp_15_r20/external/vulkan-validation-layers/tests/
H A Dvklayertests_descriptor_renderpass_framebuffer.cpp689 …VkDeviceSize scratch_buffer_size = std::max(bot_level_as.build_scratch_memory_requirements().memor… in TEST_F() local