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()
719 const Operand& inputOperand = *helper.getSubgraphInputOperand(bodyModelOperand, i); in validateWhileOperation() local721 NN_RET_CHECK(compatible(inputOperand, outputOperand)); in validateWhileOperation()1389 auto inputOperand = operands[inputIndexes[0]]; in validateOperation() local1391 auto inputType = inputOperand.type; in validateOperation()1426 if (inputOperand.dimensions.size() != 0 && outputOperand.dimensions.size() != 0 && in validateOperation()1428 inputOperand.dimensions != outputOperand.dimensions) { in validateOperation()
1764 const Operand& inputOperand = getInputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation() local1766 NN_TRY(compatible(inputOperand, outputOperand)); in validateWhileOperation()2385 auto inputOperand = operands[inputIndexes[0]]; in validateOperationButNotOperandsImpl() local2387 auto inputType = inputOperand.type; in validateOperationButNotOperandsImpl()2422 NN_VALIDATE(inputOperand.dimensions.empty() || outputOperand.dimensions.empty() || in validateOperationButNotOperandsImpl()2424 inputOperand.dimensions == outputOperand.dimensions); in validateOperationButNotOperandsImpl()