Searched defs:tensor (Results 1 – 4 of 4) sorted by relevance
| /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 | UnidirectionalSequenceLSTM.cpp | 86 inline bool hasTensor(IOperationExecutionContext* context, const uint32_t tensor) { in hasTensor()
|