Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ | |
convolution_thunk.h | 93 const BufferAllocation::Slice scratch_buffer_; member in class:xla::gpu::ConvolutionThunk |
/external/tensorflow/tensorflow/contrib/lite/kernels/ | |
optional_tensor_test.cc | 98 scratch_buffer_ = AddOutput(TensorType_FLOAT32); 238 int scratch_buffer_; member in class:tflite::__anon39267::LSTMOpModel |
lstm_test.cc | 100 scratch_buffer_ = AddOutput(TensorType_FLOAT32); 236 int scratch_buffer_; member in class:tflite::__anon39264::LSTMOpModel [all...] |
unidirectional_sequence_lstm_test.cc | 103 scratch_buffer_ = AddOutput(TensorType_FLOAT32); 241 int scratch_buffer_; member in class:tflite::__anon39285::UnidirectionalLSTMOpModel [all...] |
/frameworks/ml/nn/common/operations/ | |
LSTM.h | 143 RunTimeOperandInfo *scratch_buffer_; member in class:android::nn::LSTMCell |