Searched defs:input_shape (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Tile.cpp | 39 std::pair<int, int> TileOneDimension(const Shape& input_shape, const T* in_data, in TileOneDimension()
|
| D | LSHProjectionTest.cpp | 44 std::initializer_list<uint32_t> input_shape, in LSHProjectionOpModel()
|
| D | LSTM.cpp | 363 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMEvalFloat32()
479 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape, in LSTMEvalFloat16()
723 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMStep()
|