Searched refs:scratch_buffer_size (Results 1 – 4 of 4) sorted by relevance
329 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local330 scratch_buffer_size->data[0] = n_batch; in Prepare()333 scratch_buffer_size->data[1] = n_cell * 3; in Prepare()336 scratch_buffer_size->data[1] = n_cell * 4; in Prepare()339 scratch_buffer_size)); in Prepare()
405 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local406 scratch_buffer_size->data[0] = n_batch; in Prepare()409 scratch_buffer_size->data[1] = n_cell * 3; in Prepare()412 scratch_buffer_size->data[1] = n_cell * 4; in Prepare()415 scratch_buffer_size)); in Prepare()
1003 int scratch_buffer_size = 0; in FfMultiExp() local1040 &scratch_buffer_size); in FfMultiExp()1046 scratch_buffer = (OctStr)SAFE_ALLOC(scratch_buffer_size); in FfMultiExp()1122 int scratch_buffer_size = 0; in FfMultiExpBn() local1146 &scratch_buffer_size); in FfMultiExpBn()1152 scratch_buffer = (OctStr)SAFE_ALLOC(scratch_buffer_size); in FfMultiExpBn()
347 size_t scratch_buffer_size = in InitScratchBuffers() local350 Allocator::kAllocatorAlignment, scratch_buffer_size); in InitScratchBuffers()357 se::DeviceMemoryBase(scratch_buffer, scratch_buffer_size)); in InitScratchBuffers()