| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| basic_rnn_test.cc | 128 : batches_(batches), units_(units), input_size_(size) {
138 BuildInterpreter({{batches_, input_size_},
139 {units_, input_size_},
172 int input_size() { return input_size_; }
186 int input_size_; member in class:tflite::__anon39244::RNNOpModel
|
| unidirectional_sequence_rnn_test.cc | 132 input_size_(size) {
145 BuildInterpreter({{sequence_len_, batches_, input_size_},
146 {units_, input_size_},
150 BuildInterpreter({{batches_, sequence_len_, input_size_},
151 {units_, input_size_},
185 int input_size() { return input_size_; }
201 int input_size_; member in class:tflite::__anon39286::UnidirectionalRNNOpModel
|
| svdf_test.cc | 134 input_size_(input_size),
147 {batches_, input_size_}, // Input tensor
148 {units_ * rank, input_size_}, // weights_feature tensor
181 int input_size() { return input_size_; }
195 int input_size_; member in class:tflite::__anon39280::SVDFOpModel
|
| softmax_test.cc | 35 : batches_(batches), input_size_(size), beta_(beta) {
40 BuildInterpreter({{batches_, input_size_}});
58 int input_size_; member in class:tflite::__anon39274::SoftmaxOpModel
|
| bidirectional_sequence_rnn_test.cc | 761 int input_size_; member in class:tflite::__anon39246::BidirectionalRNNOpModel
[all...] |
| fully_connected_test.cc | 129 input_size_ = total_input_size / batches_;
133 AddInput({input.type, {units_, input_size_}, input.min, input.max});
155 int input_size() { return input_size_; }
167 int input_size_; member in class:tflite::__anon39254::BaseFullyConnectedOpModel
|
| /frameworks/ml/nn/common/operations/ |
| RNNTest.cpp | 155 input_size_(size),
159 OperandType InputTy(Type::TENSOR_FLOAT32, {batches_, input_size_});
161 OperandType WeightTy(Type::TENSOR_FLOAT32, {units_, input_size_});
178 Input_.insert(Input_.end(), batches_ * input_size_, 0.f);
211 uint32_t input_size() const { return input_size_; }
253 const uint32_t input_size_; member in class:android::nn::wrapper::BasicRNNOpModel
|
| SVDFTest.cpp | 190 input_size_(input_size),
194 {batches_, input_size_}, // Input tensor
195 {units_ * rank_, input_size_}, // weights_feature tensor
233 Input_.insert(Input_.end(), batches_ * input_size_, 0.f);
317 int input_size() const { return input_size_; }
326 const uint32_t input_size_; member in class:android::nn::wrapper::SVDFOpModel
|
| /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
|
| lstm_ops.h | 98 input_size_(input_size),
103 int input_size() const { return input_size_; }
132 return {batch_size_, input_size_};
136 return {0, input_size_};
145 const int input_size_;
|
| lstm_ops_gpu.cu.cc | 374 cell_size_, input_size_); \
|
| /toolchain/binutils/binutils-2.27/gold/ |
| merge.h | 405 merged_strings_lists_(), input_count_(0), input_size_(0)
509 size_t input_size_; member in class:gold::Output_merge_string
|
| merge.cc | 523 this->input_size_ += i;
654 program_name, buf, this->input_size_);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| conv_ops_test.py | [all...] |
| pooling_ops_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| fused_conv2d_bias_activation_op_test.py | [all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1171 int input_size_; member in class:perftools::gputools::cuda::CudnnRnnDescriptor
[all...] |
