Searched refs:inputOperand (Results 1 – 2 of 2) 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()
833 const Operand& inputOperand = *helper.getSubgraphInputOperand(bodyModelOperand, i); in validateWhileOperation() local835 NN_RET_CHECK(compatible(inputOperand, outputOperand)); in validateWhileOperation()1536 auto inputOperand = operands[inputIndexes[0]]; in validateOperation() local1538 auto inputType = inputOperand.type; in validateOperation()1573 if (inputOperand.dimensions.size() != 0 && outputOperand.dimensions.size() != 0 && in validateOperation()1575 inputOperand.dimensions != outputOperand.dimensions) { in validateOperation()