OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
full:scratch_buffer_size
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc
288
TfLiteIntArray*
scratch_buffer_size
= TfLiteIntArrayCreate(2);
local
289
scratch_buffer_size
->data[0] = n_batch;
291
scratch_buffer_size
->data[1] = n_cell * 3;
293
scratch_buffer_size
));
295
TfLiteIntArray*
scratch_buffer_size
= TfLiteIntArrayCreate(2);
local
296
scratch_buffer_size
->data[0] = n_batch;
298
scratch_buffer_size
->data[1] = n_cell * 4;
300
scratch_buffer_size
));
unidirectional_sequence_lstm.cc
290
TfLiteIntArray*
scratch_buffer_size
= TfLiteIntArrayCreate(2);
local
291
scratch_buffer_size
->data[0] = n_batch;
293
scratch_buffer_size
->data[1] = n_cell * 3;
295
scratch_buffer_size
));
297
TfLiteIntArray*
scratch_buffer_size
= TfLiteIntArrayCreate(2);
local
298
scratch_buffer_size
->data[0] = n_batch;
300
scratch_buffer_size
->data[1] = n_cell * 4;
302
scratch_buffer_size
));
/external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc
290
size_t
scratch_buffer_size
= Eigen::kCudaScratchSize + sizeof(unsigned int);
local
292
Allocator::kAllocatorAlignment,
scratch_buffer_size
);
301
scratch_buffer_size
));
[
all
...]
Completed in 175 milliseconds