| /external/tensorflow/tensorflow/lite/kernels/ |
| log_softmax_test.cc | 35 : batches_(batches), input_size_(size) {
40 BuildInterpreter({{batches_, input_size_}});
58 int input_size_; member in class:tflite::__anon46058::LogSoftmaxOpModel
|
| softmax_test.cc | 35 : batches_(batches), input_size_(size), beta_(beta) {
40 BuildInterpreter({{batches_, input_size_}});
58 int input_size_; member in class:tflite::__anon46096::SoftmaxOpModel
|
| basic_rnn_test.cc | 179 : batches_(batches), units_(units), input_size_(size) {
189 BuildInterpreter({{batches_, input_size_}, // input tensor
190 {units_, input_size_}, // weights tensor
216 int input_size() { return input_size_; }
230 int input_size_; member in class:tflite::__anon45998::RNNOpModel
|
| unidirectional_sequence_rnn_test.cc | 181 input_size_(size) {
194 BuildInterpreter({{sequence_len_, batches_, input_size_},
195 {units_, input_size_},
200 BuildInterpreter({{batches_, sequence_len_, input_size_},
201 {units_, input_size_},
228 int input_size() { return input_size_; }
244 int input_size_; member in class:tflite::__anon46118::UnidirectionalRNNOpModel
|
| svdf_test.cc | 137 input_size_(input_size),
153 {batches_, input_size_}, // input tensor
154 {units_ * rank, input_size_}, // weights_feature tensor
179 int input_size() { return input_size_; }
193 int input_size_; member in class:tflite::__anon46109::BaseSVDFOpModel
|
| bidirectional_sequence_rnn_test.cc | 780 int input_size_; member in class:tflite::__anon46002::BidirectionalRNNOpModel
[all...] |
| fully_connected_test.cc | 148 input_size_ = total_input_size / batches_;
152 AddInput({input.type, {units_, input_size_}, input.min, input.max});
183 int input_size() { return input_size_; }
195 int input_size_; member in class:tflite::__anon46032::BaseFullyConnectedOpModel
284 input_size_ = total_input_size / batches_;
315 int input_size() { return input_size_; }
327 int input_size_; member in class:tflite::__anon46032::HybridFullyConnectedOpModel
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| gru_ops.h | 33 input_size_(input_size),
39 return {batch_size_, input_size_};
43 return {0, input_size_};
64 const int input_size_; member in struct:tensorflow::functor::GRUCell
|
| /external/tensorflow/tensorflow/core/kernels/ |
| cudnn_rnn_ops.cc | 155 input_size_(input_size),
184 std::to_string(input_size_),
202 return std::make_tuple(num_layers_, input_size_, num_units_, seq_length_,
208 const int input_size_; member in class:tensorflow::TFRNNInputMode::__anon45229::CudnnRnnParameters
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | 627 : batches_(batches), input_size_(size), beta_(beta) {
632 BuildInterpreter({{batches_, input_size_}});
650 int input_size_; member in class:tflite::__anon45929::SoftmaxOpModel
1957 int input_size_; member in class:tflite::__anon45929::RNNOpModel
2151 int input_size_; member in class:tflite::__anon45929::BaseSVDFOpModel
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1138 int input_size_; member in class:stream_executor::gpu::CudnnRnnDescriptor
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.cc | 1751 int input_size_; member in class:stream_executor::gpu::MIOpenRnnDescriptor
[all...] |
