Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DTile.cpp39 std::pair<int, int> TileOneDimension(const Shape& input_shape, const T* in_data, in TileOneDimension() argument
41 const int dimension_size = input_shape.dimensions[dimension]; in TileOneDimension()
42 if (dimension == input_shape.dimensions.size() - 1) { in TileOneDimension()
53 input_shape, copy_from_data, multipliers, copy_to_data, dimension + 1); in TileOneDimension()
DLSTM.cpp363 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMEvalFloat32() argument
386 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32()
390 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat32()
391 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32()
392 : getSizeOfDimension(input_shape, 0); in LSTMEvalFloat32()
393 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat32()
397 Shape batchInputShape = input_shape; in LSTMEvalFloat32()
410 transposeFirstTwoDimensions<float>(input_buffer, input_shape, transposedInput.data()); in LSTMEvalFloat32()
413 transposeFirstTwoDimensions<float>(aux_input_buffer, input_shape, in LSTMEvalFloat32()
416 transposeFirstTwoDimensions(input_shape, &transposedInputShape); in LSTMEvalFloat32()
[all …]
DLSTM.h108 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
133 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape,
161 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
DLSHProjectionTest.cpp44 std::initializer_list<uint32_t> input_shape, in LSHProjectionOpModel() argument
51 OperandType InputTy(Type::TENSOR_INT32, input_shape); in LSHProjectionOpModel()