Home
last modified time | relevance | path

Searched refs:input_shape (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp411 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMEvalFloat32() argument
434 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32()
438 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat32()
439 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32()
440 : getSizeOfDimension(input_shape, 0); in LSTMEvalFloat32()
441 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat32()
445 Shape batchInputShape = input_shape; in LSTMEvalFloat32()
458 transposeFirstTwoDimensions<float>(input_buffer, input_shape, transposedInput.data()); in LSTMEvalFloat32()
461 transposeFirstTwoDimensions<float>(aux_input_buffer, input_shape, in LSTMEvalFloat32()
464 transposeFirstTwoDimensions(input_shape, &transposedInputShape); in LSTMEvalFloat32()
[all …]
DTile.cpp43 std::pair<int, int> TileOneDimension(const Shape& input_shape, const T* in_data, in TileOneDimension() argument
45 const int dimension_size = input_shape.dimensions[dimension]; in TileOneDimension()
46 if (static_cast<size_t>(dimension) == input_shape.dimensions.size() - 1) { in TileOneDimension()
57 input_shape, copy_from_data, multipliers, copy_to_data, dimension + 1); in TileOneDimension()
DLSHProjectionTest.cpp45 std::initializer_list<uint32_t> input_shape, in LSHProjectionOpModel() argument
52 OperandType InputTy(Type::TENSOR_INT32, input_shape); in LSHProjectionOpModel()
/packages/modules/NeuralNetworks/common/types/operations/include/
DLSTM.h107 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
132 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape,
160 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,