HomeSort by relevance Sort by last modified time
    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