Searched refs:spatialSize (Results 1 – 8 of 8) sorted by relevance
73 val spatialSize = img.shape(2) * img.shape(3) constant74 val r = clip(datas.take(spatialSize).map(_ + 123.68f))75 val g = clip(datas.drop(spatialSize).take(spatialSize).map(_ + 116.779f))76 val b = clip(datas.takeRight(spatialSize).map(_ + 103.939f))77 val pixels = for (i <- 0 until spatialSize)
87 val spatialSize = img.shape(2) * img.shape(3) constant88 val r = clip(datas.take(spatialSize).map(_ + 123.68f))89 val g = clip(datas.drop(spatialSize).take(spatialSize).map(_ + 116.779f))90 val b = clip(datas.takeRight(spatialSize).map(_ + 103.939f))91 val pixels = for (i <- 0 until spatialSize)
117 int spatialSize = total(inpShape, endAxis); // spatialSize != 1 in forward() local124 Mat inpSlice(1, spatialSize, CV_32F, inpData); in forward()125 Mat outSlice(1, spatialSize, CV_32F, outData); in forward()127 inpData += spatialSize; in forward()128 outData += spatialSize; in forward()
1352 size_t spatialSize = 1; in runChannelAxisTest() local1354 spatialSize *= shape[x]; in runChannelAxisTest()1366 for (size_t i = 0; i < spatialSize; ++i) { in runChannelAxisTest()
822 int spatialSize = 1; in TEST_P() local824 spatialSize *= inpShape[i]; in TEST_P()827 float w = weightsData[(i / spatialSize) % weights.total()]; in TEST_P()