Lines Matching defs:inputDimensions
329 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in argMinMaxTest() local
364 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in dequantizeOpTest() local
384 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in expandDimsTest() local
406 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in gatherTest() local
432 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in quantizeOpTest() local
509 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in splitTest() local
537 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in tileTest() local
560 uint32_t inputDimensions[4] = {4, 5, 6, 7}; in topkV2Test() local
579 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in simpleMathOpTest() local
639 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in TEST() local
658 uint32_t inputDimensions[] = {2, 2, 2, 2, 2}; in binaryOpTest() local
701 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in activationOpTest() local
843 uint32_t inputDimensions[3] = {2, 3, 4}; in reshapeOpTest() local
860 uint32_t inputDimensions[3] = {2, 2, 2}; in logSoftmaxOpTest() local
898 uint32_t inputDimensions[3] = {2, 2, 2}; in meanOpTest() local
922 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in padOpTest() local
941 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in padV2OpTest() local
965 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in softmaxOpTest() local
995 uint32_t inputDimensions[4] = {2, 4, 4, 2}; in poolingOpTest() local
1080 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in spaceDepthOpTest() local
1127 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in spaceBatchOpTest() local
1185 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in transposeAndSqueezeOpTest() local
1225 uint32_t inputDimensions[4] = {2, 4, 4, 2}; in convOpTest() local
1348 uint32_t inputDimensions[4] = {1, 2, 2, 2}; in depthwiseConvOpTest() local
1472 uint32_t inputDimensions[2] = {5, 5}; in fullyConnectedOpTest() local
1513 uint32_t inputDimensions[2] = {5, 5}; in concatenationTest() local
1546 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in resizeBilinearOpTest() local
1578 uint32_t inputDimensions[2] = {5, 5}; in embeddingLookupTest() local
1633 uint32_t inputDimensions[2] = {5, 5}; in lshProjectionTest() local
2223 uint32_t inputDimensions[2] = {5, 5}; in stridedSliceOpTest() local
2696 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in logicalTest() local
2718 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in comparisonTest() local
2781 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in reduceOpTest() local
2840 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in selectTest() local
2858 const uint32_t inputDimensions[] = {3, 3}; in powTest() local
2907 uint32_t inputDimensions[3] = {2, 2, 2}; in castOpTest() local