Lines Matching refs:output
28 std::vector<float> output(kVectorSize);
29 ClipVector(input, kVectorSize, kAbsLimit, output.data());
30 EXPECT_THAT(output,
44 std::vector<float> output(kRow * kBatch);
45 std::fill(output.begin(), output.end(), 3.0);
47 output.data(), /*result_stride=*/1);
48 EXPECT_THAT(output, ElementsAreArray(ArrayFloatNear({1., 5., 13., //
67 std::vector<float> output(kVectorSize);
68 VectorVectorCwiseProduct(input1, input2, kVectorSize, output.data());
69 EXPECT_THAT(output,
80 std::vector<float> output(kVectorSize);
81 std::fill(output.begin(), output.end(), 1.0);
83 output.data());
84 EXPECT_THAT(output,
93 std::vector<float> output(kVectorSize * kBatchSize);
94 VectorBatchVectorAssign(input, kVectorSize, kBatchSize, output.data());
95 EXPECT_THAT(output, ElementsAreArray(ArrayFloatNear(
103 std::vector<float> output(kVectorSize);
104 ApplySigmoidToVector(input, kVectorSize, output.data());
105 EXPECT_THAT(output, ElementsAreArray(ArrayFloatNear(
112 std::vector<float> output(kVectorSize);
113 ApplyActivationToVector(input, kVectorSize, kTfLiteActRelu, output.data());
114 EXPECT_THAT(output,
117 ApplyActivationToVector(input, kVectorSize, kTfLiteActTanh, output.data());
118 EXPECT_THAT(output, ElementsAreArray(ArrayFloatNear(
125 std::vector<float> output(kVectorSize);
126 CopyVector(input, kVectorSize, output.data());
127 EXPECT_THAT(output,
134 std::vector<float> output(kVectorSize);
135 Sub1Vector(input, kVectorSize, output.data());
136 EXPECT_THAT(output,
142 std::vector<float> output(kVectorSize);
143 ZeroVector(output.data(), kVectorSize);
144 EXPECT_THAT(output,
155 std::vector<float> output(kBatch);
157 output.data(), /*result_stride=*/1);
158 EXPECT_THAT(output, ElementsAreArray(ArrayFloatNear({0.5, 1.75})));
