Searched refs:inputOperand (Results 1 – 3 of 3) sorted by relevance
148 const auto& inputOperand = model.main.operands[model.main.inputIndexes[0]]; in expectMultinomialDistributionWithinTolerance() local150 ASSERT_EQ(inputOperand.dimensions.size(), 2u); in expectMultinomialDistributionWithinTolerance()153 const int kBatchSize = inputOperand.dimensions[0]; in expectMultinomialDistributionWithinTolerance()154 const int kNumClasses = inputOperand.dimensions[1]; in expectMultinomialDistributionWithinTolerance()164 const uint32_t inputLength = getNumberOfElements(inputOperand); in expectMultinomialDistributionWithinTolerance()166 if (inputOperand.type == TestOperandType::TENSOR_FLOAT32) { in expectMultinomialDistributionWithinTolerance()167 const float* inputRaw = inputOperand.data.get<float>(); in expectMultinomialDistributionWithinTolerance()169 } else if (inputOperand.type == TestOperandType::TENSOR_FLOAT16) { in expectMultinomialDistributionWithinTolerance()170 const _Float16* inputRaw = inputOperand.data.get<_Float16>(); in expectMultinomialDistributionWithinTolerance()
687 const Operand& inputOperand = *helper.getSubgraphInputOperand(bodyModelOperand, i); in validateWhileOperation() local689 NN_RET_CHECK(compatible(inputOperand, outputOperand)); in validateWhileOperation()1361 auto inputOperand = operands[inputIndexes[0]]; in validateOperation() local1363 auto inputType = inputOperand.type; in validateOperation()1398 if (inputOperand.dimensions.size() != 0 && outputOperand.dimensions.size() != 0 && in validateOperation()1400 inputOperand.dimensions != outputOperand.dimensions) { in validateOperation()
1719 const Operand& inputOperand = getInputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation() local1721 NN_TRY(compatible(inputOperand, outputOperand)); in validateWhileOperation()