| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | OpContext.h | 36 std::vector<int64_t> input_size_; variable
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_tester.h | 79 int32_t input_size_ = 1; variable
|
| D | reduce_tester.h | 109 int32_t input_size_; variable
|
| D | fully_connected_tester.h | 143 int32_t input_size_ = 1; variable
|
| D | quantized_reduce_tester.h | 143 int32_t input_size_; variable
|
| D | quantized_fully_connected_tester.h | 174 int32_t input_size_ = 1; variable
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | log_softmax_test.cc | 58 int input_size_; member in tflite::__anon993bee410111::LogSoftmaxOpModel
|
| D | softmax_test.cc | 59 int input_size_; member in tflite::__anon676cc4200111::SoftmaxOpModel
|
| D | svdf_test.cc | 194 int input_size_; member in tflite::__anon7b3953510111::BaseSVDFOpModel
531 int input_size_; member in tflite::__anon7b3953510111::IntegerSVDFOpModel
|
| D | quant_basic_lstm_test.cc | 156 int input_size_; member in tflite::__anonc69a0ae70111::QuantizedLSTMOpModel
|
| D | basic_rnn_test.cc | 229 int input_size_; member in tflite::__anond471c92d0111::RNNOpModel
|
| D | batch_matmul_test.cc | 440 int input_size_; member in tflite::__anonf39c542f0111::HybridBatchMatMulOpModel
972 int input_size_; member in tflite::__anonf39c542f0111::QuantizedBatchMatMulOpModel
|
| D | unidirectional_sequence_rnn_test.cc | 245 int input_size_; member in tflite::__anon8708873d0111::UnidirectionalRNNOpModel
|
| D | fully_connected_test.cc | 256 int input_size_; member in tflite::__anon395836bc0111::BaseFullyConnectedOpModel
463 int input_size_; member in tflite::__anon395836bc0111::HybridFullyConnectedOpModel
1608 int input_size_; member in tflite::__anon395836bc0111::SparseFullyConnectedOpModel
|
| D | lstm_eval_test.cc | 151 std::vector<int32_t> input_size_ = {n_batch_, n_input_}; member in tflite::__anondd2239e50111::BaseLstmParam
|
| D | bidirectional_sequence_rnn_test.cc | 843 int input_size_; member in tflite::__anonb5fd7e9c0111::BidirectionalRNNOpModel
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | gru_ops.h | 64 const int input_size_; member
|
| D | lstm_ops.h | 156 const int input_size_; member
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | matmul_test.cc | 103 int input_size_; member in tflite::FullyConnectedOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_signed_quantization_test.cc | 129 int input_size_; member in tflite::__anon5ec1492c0111::HybridFullyConnectedOpModel
|
| D | nnapi_delegate_test.cc | 2868 int input_size_; member in tflite::__anon8716bded0111::RNNOpModel
3061 int input_size_; member in tflite::__anon8716bded0111::BaseSVDFOpModel
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 211 const int input_size_; member in tensorflow::__anonf4c408700111::CudnnRnnParameters
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1886 int input_size_; member in stream_executor::gpu::MIOpenRnnDescriptor
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1411 int input_size_; member in stream_executor::gpu::CudnnRnnDescriptor
|