Searched refs:kTensorSize (Results 1 – 4 of 4) sorted by relevance
35 static const int kTensorSize = 8 * 8 * 3;75 NSMutableData* inputData = [NSMutableData dataWithLength:sizeof(float) * kTensorSize];76 for (int i = 0; i < kTensorSize / kTensorChannels; ++i) {98 float output[kTensorSize];99 [outputData getBytes:output length:(sizeof(float) * kTensorSize)];100 for (int i = 0; i < kTensorSize / kTensorChannels; ++i) {
40 static const int kTensorSize = 8 * 8 * 3;82 NSMutableData* inputData = [NSMutableData dataWithLength:sizeof(float) * kTensorSize];83 for (int i = 0; i < kTensorSize / kTensorChannels; ++i) {108 float output[kTensorSize];110 [outputData getBytes:output length:(sizeof(float) * kTensorSize)];111 for (int i = 0; i < kTensorSize / kTensorChannels; ++i) {
1059 const int kTensorSize = 1000; in TEST_F() local1061 OpContext op_context = DescribeUnaryOp(unary_op.first, kTensorSize); in TEST_F()1065 unary_op.second * kTensorSize / in TEST_F()2041 const int kTensorSize = 1000; in TEST_F() local2043 OpContext op_context = DescribeUnaryOp(identity_op, kTensorSize); in TEST_F()2053 EXPECT_EQ(cost.max_memory, kTensorSize * 4); in TEST_F()2072 const int kTensorSize = 1000; in TEST_F() local2074 OpContext op_context = DescribeUnaryOp(reshape_op, kTensorSize); in TEST_F()2084 EXPECT_EQ(cost.max_memory, kTensorSize * 4); in TEST_F()
94 constexpr int kTensorSize = 16; in TEST() local97 {kTensorSize}, quant, in TEST()103 for (int i = 0; i < kTensorSize; ++i) { in TEST()112 constexpr int kTensorSize = 16; in TEST() local117 {kTensorSize}, quant, in TEST()123 for (int i = 0; i < kTensorSize; ++i) { in TEST()