Searched refs:kVecWidth (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 247 const int kVecWidth = 4; in DescribeFusedConv2DBiasActivation() local
262 EXPECT_EQ(iz1 % kVecWidth, 0); in DescribeFusedConv2DBiasActivation()
263 DescribeTensor5D(batch, iz1 / kVecWidth, ix, iy, kVecWidth, in DescribeFusedConv2DBiasActivation()
272 EXPECT_EQ(iz2 % kVecWidth, 0); in DescribeFusedConv2DBiasActivation()
274 DescribeTensor5D(oz, iz2 / kVecWidth, kx, ky, kVecWidth, in DescribeFusedConv2DBiasActivation()
289 EXPECT_EQ(oz % kVecWidth, 0); in DescribeFusedConv2DBiasActivation()
290 DescribeTensor5D(batch, oz / kVecWidth, ox, oy, kVecWidth, side_input); in DescribeFusedConv2DBiasActivation()
|