Lines Matching refs:outputDimensions

1048     uint32_t outputDimensions[3] = {2, 2, 2};  in argMinMaxTest()  local
1052 .dimensions = outputDimensions, in argMinMaxTest()
1107 uint32_t outputDimensions[5] = {2, 2, 2, 2, 2}; in expandDimsTest() local
1108 ANeuralNetworksOperandType output = getOpType(inputOperandType, 5, outputDimensions); in expandDimsTest()
1135 uint32_t outputDimensions[7] = {2, 2, 2, 2, 2, 2, 2}; in gatherTest() local
1136 ANeuralNetworksOperandType output = getOpType(inputOperandType, 7, outputDimensions); in gatherTest()
1241 uint32_t outputDimensions[2] = {2, 2}; in splitTest() local
1242 ANeuralNetworksOperandType output0 = getOpType(inputOperandType, 2, outputDimensions); in splitTest()
1243 ANeuralNetworksOperandType output1 = getOpType(inputOperandType, 2, outputDimensions); in splitTest()
1266 uint32_t outputDimensions[8] = {2, 2, 2, 2, 2, 2, 2, 2}; in tileTest() local
1267 ANeuralNetworksOperandType output0 = getOpType(inputOperandType, 8, outputDimensions); in tileTest()
1285 uint32_t outputDimensions[4] = {4, 5, 6, 3}; in topkV2Test() local
1286 ANeuralNetworksOperandType outputValues = getOpType(inputOperandType, 4, outputDimensions); in topkV2Test()
1288 getOpType(ANEURALNETWORKS_TENSOR_INT32, 4, outputDimensions); in topkV2Test()
1682 uint32_t outputDimensions[2] = {4, 6}; in reshapeOpTest() local
1683 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 2, outputDimensions); in reshapeOpTest()
1776 uint32_t outputDimensions[4] = {4, 3, 4, 3}; in padOpTest() local
1777 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outputDimensions); in padOpTest()
1817 uint32_t outputDimensions[4] = {4, 3, 4, 3}; in padV2OpTest() local
1818 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outputDimensions); in padV2OpTest()
4458 uint32_t outputDimensions[3] = {3, 4, 5}; in fillTest() local
4459 ANeuralNetworksOperandType output = getOpType(outputOperandType, 3, outputDimensions); in fillTest()
4511 uint32_t outputDimensions[3] = {1, 2, 4}; in batchMatmulTest() local
4512 ANeuralNetworksOperandType output = getOpType(operandType, 3, outputDimensions); in batchMatmulTest()
4735 uint32_t outputDimensions[outputRank]; in packTest() local
4738 outputDimensions[outDim] = inputTensorCount; in packTest()
4740 outputDimensions[outDim] = inputDimensions[inDim++]; in packTest()
4744 getOpType(operandCode, outputRank, outputDimensions); in packTest()
4784 uint32_t outputDimensions[outputRank]; in packTestBadQuantization() local
4785 std::copy(std::begin(inputDimensions), std::end(inputDimensions), std::begin(outputDimensions)); in packTestBadQuantization()
4786 outputDimensions[inputRank] = inputTensorCount; in packTestBadQuantization()
4797 getOpType(operandCode, outputRank, outputDimensions); in packTestBadQuantization()
4845 uint32_t outputDimensions[outputRank]; in packTestBadRank() local
4847 std::begin(outputDimensions) + 1); in packTestBadRank()
4848 outputDimensions[0] = inputTensorCount; in packTestBadRank()
4881 getOpType(operandCode, outputRank, outputDimensions); in packTestBadRank()