| /external/tensorflow/tensorflow/lite/kernels/ |
| D | svdf_test.cc | 137 input_size_(input_size), in BaseSVDFOpModel()
153 {batches_, input_size_}, // input tensor in BaseSVDFOpModel()
154 {units_ * rank, input_size_}, // weights_feature tensor in BaseSVDFOpModel()
179 int input_size() { return input_size_; } in input_size()
193 int input_size_; member in tflite::__anond1a0a9e40111::BaseSVDFOpModel
|
| D | unidirectional_sequence_rnn_test.cc | 181 input_size_(size) { in UnidirectionalRNNOpModel()
194 BuildInterpreter({{sequence_len_, batches_, input_size_}, in UnidirectionalRNNOpModel()
195 {units_, input_size_}, in UnidirectionalRNNOpModel()
200 BuildInterpreter({{batches_, sequence_len_, input_size_}, in UnidirectionalRNNOpModel()
201 {units_, input_size_}, in UnidirectionalRNNOpModel()
228 int input_size() { return input_size_; } in input_size()
244 int input_size_; member in tflite::__anon2f8e60700111::UnidirectionalRNNOpModel
|
| D | basic_rnn_test.cc | 179 : batches_(batches), units_(units), input_size_(size) { in RNNOpModel()
189 BuildInterpreter({{batches_, input_size_}, // input tensor in RNNOpModel()
190 {units_, input_size_}, // weights tensor in RNNOpModel()
216 int input_size() { return input_size_; } in input_size()
230 int input_size_; member in tflite::__anon35c533a00111::RNNOpModel
|
| D | log_softmax_test.cc | 35 : batches_(batches), input_size_(size) { in LogSoftmaxOpModel()
40 BuildInterpreter({{batches_, input_size_}}); in LogSoftmaxOpModel()
58 int input_size_; member in tflite::__anon9d13c1740111::LogSoftmaxOpModel
|
| D | softmax_test.cc | 35 : batches_(batches), input_size_(size), beta_(beta) { in SoftmaxOpModel()
40 BuildInterpreter({{batches_, input_size_}}); in SoftmaxOpModel()
58 int input_size_; member in tflite::__anona7ecf5d30111::SoftmaxOpModel
|
| D | bidirectional_sequence_rnn_test.cc | 663 input_size_(input_size) { in BidirectionalRNNOpModel()
677 aux_input_size = input_size_; in BidirectionalRNNOpModel()
696 (time_major) ? std::vector<int>({sequence_len_, batches_, input_size_}) in BidirectionalRNNOpModel()
697 : std::vector<int>({batches_, sequence_len_, input_size_}); in BidirectionalRNNOpModel()
701 {fw_units_, input_size_}, // fw_weights in BidirectionalRNNOpModel()
705 {bw_units_, input_size_}, // bw_weights in BidirectionalRNNOpModel()
754 int input_size() { return input_size_; } in input_size()
780 int input_size_; member in tflite::__anon65c37d6f0111::BidirectionalRNNOpModel
|
| D | 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_; } in input_size()
195 int input_size_; member in tflite::__anon3e7c6b6f0111::BaseFullyConnectedOpModel
284 input_size_ = total_input_size / batches_;
315 int input_size() { return input_size_; } in input_size()
327 int input_size_; member in tflite::__anon3e7c6b6f0111::HybridFullyConnectedOpModel
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | gru_ops.h | 33 input_size_(input_size), in GRUCell()
39 return {batch_size_, input_size_}; in x_extends()
43 return {0, input_size_}; in h_offsets()
64 const int input_size_; member
|
| D | lstm_ops.h | 97 input_size_(input_size), in LSTMBlockCell()
102 int input_size() const { return input_size_; } in input_size()
131 return {batch_size_, input_size_}; in xh_x_extents()
135 return {0, input_size_}; in xh_h_offsets()
144 const int input_size_; member
|
| D | lstm_ops_gpu.cu.cc | 426 h, batch_size_, cell_size_, input_size_); \
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 627 : batches_(batches), input_size_(size), beta_(beta) { in SoftmaxOpModel()
632 BuildInterpreter({{batches_, input_size_}}); in SoftmaxOpModel()
650 int input_size_; member in tflite::__anon6d004b400111::SoftmaxOpModel
1906 : batches_(batches), units_(units), input_size_(size) { in RNNOpModel()
1916 BuildInterpreter({{batches_, input_size_}, // input tensor in RNNOpModel()
1917 {units_, input_size_}, // weights tensor in RNNOpModel()
1943 int input_size() { return input_size_; } in input_size()
1957 int input_size_; member in tflite::__anon6d004b400111::RNNOpModel
2091 input_size_(input_size), in BaseSVDFOpModel()
2109 {batches_, input_size_}, // input tensor in BaseSVDFOpModel()
[all …]
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test_base.py | 598 for index, (input_size_, filter_size_, output_size_, stride_,
601 _GetInceptionFwdTest(input_size_, filter_size_, stride_, padding_))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 1956 for (name_, input_size_, filter_size_, output_size_, stride_,
1959 GetMaxPoolFwdTest(input_size_, filter_size_, stride_, padding_))
1961 GetMaxPoolGradTest(input_size_, filter_size_, output_size_, stride_,
1964 GetMaxPoolGradGradTest(input_size_, filter_size_, output_size_,
|
| D | conv_ops_test.py | 2975 for index, (input_size_, filter_size_, output_size_, stride_,
2979 GetInceptionFwdTest(input_size_, filter_size_, stride_,
2984 input_size_, filter_size_, stride_, padding_)))
2987 GetInceptionBackInputTest(input_size_, filter_size_,
2991 GetInceptionBackFilterTest(input_size_, filter_size_,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 155 input_size_(input_size), in CudnnRnnParameters()
184 std::to_string(input_size_), in ToString()
202 return std::make_tuple(num_layers_, input_size_, num_units_, seq_length_, in get_data_as_tuple()
208 const int input_size_; member in tensorflow::__anon454a70830111::CudnnRnnParameters
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1017 input_size_(input_size), in CudnnRnnDescriptor()
1110 int input_size() const { return input_size_; } in input_size()
1138 int input_size_; member in stream_executor::gpu::CudnnRnnDescriptor
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1691 input_size_(input_size), in MIOpenRnnDescriptor()
1726 int input_size() const { return input_size_; } in input_size()
1751 int input_size_; member in stream_executor::gpu::MIOpenRnnDescriptor
|