Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/micro/memory_planner/
Dgreedy_memory_planner_test.cc264 constexpr int scratch_buffer_size = 40; in TF_LITE_MICRO_TEST() local
Dgreedy_memory_planner.cc42 int scratch_buffer_size) in GreedyMemoryPlanner()
/external/tensorflow/tensorflow/lite/kernels/
Dbatch_matmul.cc158 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(lhs_rank); in InitializeTemporaries() local
183 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(rhs_rank); in InitializeTemporaries() local
Dunidirectional_sequence_lstm.cc962 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
1188 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
Dlstm.cc1418 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
1772 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
1810 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc407 size_t scratch_buffer_size = Eigen::kGpuScratchSize + sizeof(unsigned int); in InitScratchBuffers() local
/external/vulkan-validation-layers/tests/
Dvklayertests_descriptor_renderpass_framebuffer.cpp689 …VkDeviceSize scratch_buffer_size = std::max(bot_level_as.build_scratch_memory_requirements().memor… in TEST_F() local