Searched defs:tensor (Results 1 – 8 of 8) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_rnn_1_3.mod.py | 20 def convert_to_time_major(tensor, tensor_shape): argument
30 def reverse_batch_major(tensor, tensor_shape): argument
33 def split_tensor_in_two(tensor, tensor_shape): argument
|
| D | bidirectional_sequence_rnn_state_output.mod.py | 20 def convert_to_time_major(tensor, tensor_shape): argument
31 def reverse_batch_major(tensor, tensor_shape): argument
35 def split_tensor_in_two(tensor, tensor_shape): argument
|
| D | unidirectional_sequence_rnn.mod.py | 42 def convert_to_time_major(tensor, num_batches, max_time, input_size): argument
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | bidirectional_sequence_rnn.mod.py | 20 def convert_to_time_major(tensor, tensor_shape): argument
30 def reverse_batch_major(tensor, tensor_shape): argument
33 def split_tensor_in_two(tensor, tensor_shape): argument
|
| D | unidirectional_sequence_rnn.mod.py | 39 def convert_to_time_major(tensor, num_batches, max_time, input_size): argument
|
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTMTest.cpp | 225 Result setInputTensor(Execution* execution, int tensor, const std::vector<T>& data) { in setInputTensor()
229 Result setOutputTensor(Execution* execution, int tensor, std::vector<T>* data) { in setOutputTensor()
|
| D | QLSTM.cpp | 98 inline bool hasTensor(IOperationExecutionContext* context, const uint32_t tensor) { in hasTensor()
173 for (const int tensor : requiredTensorInputs) { in prepare() local
|
| D | UnidirectionalSequenceLSTM.cpp | 93 inline bool hasTensor(IOperationExecutionContext* context, const uint32_t tensor) { in hasTensor()
|