Home
last modified time | relevance | path

Searched refs:getScalarData (Results 1 – 7 of 7) sorted by relevance

/packages/modules/NeuralNetworks/driver/sample_hidl/
DSampleDriverFloatXNNPACK.cpp592 int activation = getScalarData<int32_t>(operands[ins[2]]); in VisitAddNode()
625 use_nchw = getScalarData<bool>(operands[ins[7]]); in VisitAveragePool2DNode()
628 use_nchw = getScalarData<bool>(operands[ins[10]]); in VisitAveragePool2DNode()
643 input_padding_left = static_cast<uint32_t>(getScalarData<int32_t>(operands[ins[1]])); in VisitAveragePool2DNode()
644 input_padding_right = static_cast<uint32_t>(getScalarData<int32_t>(operands[ins[2]])); in VisitAveragePool2DNode()
645 input_padding_top = static_cast<uint32_t>(getScalarData<int32_t>(operands[ins[3]])); in VisitAveragePool2DNode()
646 input_padding_bottom = static_cast<uint32_t>(getScalarData<int32_t>(operands[ins[4]])); in VisitAveragePool2DNode()
647 stride_width = getScalarData<int32_t>(operands[ins[5]]); in VisitAveragePool2DNode()
648 stride_height = getScalarData<int32_t>(operands[ins[6]]); in VisitAveragePool2DNode()
649 filter_width = getScalarData<int32_t>(operands[ins[7]]); in VisitAveragePool2DNode()
[all …]
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp817 int32_t blockSize = getScalarData<int32_t>(operands[ins[1]]); in executeOperation()
818 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation()
881 int32_t blockSize = getScalarData<int32_t>(operands[ins[1]]); in executeOperation()
882 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation()
999 const auto merge_outputs = getScalarData<bool>( in executeOperation()
1103 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation()
1175 bool data_layout = inCount == 4 ? getScalarData<bool>(operands[ins[3]]) : false; in executeOperation()
1262 float pad_value = isV2 ? getScalarData<float>(operands[ins[2]]) : 0; in executeOperation()
1267 _Float16 pad_value = isV2 ? getScalarData<_Float16>(operands[ins[2]]) : 0; in executeOperation()
1274 isV2 ? getScalarData<uint8_t>(operands[ins[2]]) : outShape.offset; in executeOperation()
[all …]
/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomial.cpp63 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()
80 getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Prepare()
DLSHProjection.cpp39 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()
64 auto type = static_cast<LSHProjectionType>(getScalarData<int32_t>(typeOperand)); in Prepare()
DSVDF.cpp82 const int rank = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in Prepare()
DRNN.cpp39 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]])); in RNN()
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h247 T getScalarData(const RunTimeOperandInfo& info) { in getScalarData() function
258 return getScalarData<T>(info); in getScalarDataWithDefault()