Searched refs:getScalarData (Results 1 – 6 of 6) sorted by relevance
870 int32_t blockSize = getScalarData<int32_t>(operands[ins[1]]); in executeOperation()871 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation()934 int32_t blockSize = getScalarData<int32_t>(operands[ins[1]]); in executeOperation()935 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation()1052 const auto merge_outputs = getScalarData<bool>( in executeOperation()1159 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation()1231 bool data_layout = inCount == 4 ? getScalarData<bool>(operands[ins[3]]) : false; in executeOperation()1318 float pad_value = isV2 ? getScalarData<float>(operands[ins[2]]) : 0; in executeOperation()1323 _Float16 pad_value = isV2 ? getScalarData<_Float16>(operands[ins[2]]) : 0; in executeOperation()1330 isV2 ? getScalarData<uint8_t>(operands[ins[2]]) : outShape.offset; in executeOperation()[all …]
57 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()74 getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Prepare()
40 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()65 auto type = static_cast<LSHProjectionType>(getScalarData<int32_t>(typeOperand)); in Prepare()
82 const int rank = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in Prepare()
43 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]])); in RNN()
247 T getScalarData(const RunTimeOperandInfo& info) { in getScalarData() function258 return getScalarData<T>(info); in getScalarDataWithDefault()