Searched refs:kOutWidth (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | image_grad_test.cc | 237 constexpr int kOutWidth = 6; in TEST_F() local
244 kXShape, kOutHeight, kOutWidth, scale, translation, kernel_type, in TEST_F()
253 constexpr int kOutWidth = 6; in TEST_F() local
258 TestScaleAndTranslate<float, float, float>(kXShape, kOutHeight, kOutWidth, in TEST_F()
269 constexpr int kOutWidth = 3; in TEST_F() local
276 kXShape, kOutHeight, kOutWidth, scale, translation, kernel_type, in TEST_F()
286 constexpr int kOutWidth = 3; in TEST_F() local
293 kXShape, kOutHeight, kOutWidth, scale, translation, kernel_type, in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 51 constexpr int kOutWidth = 5; variable
160 {kBatchSize, kOutHeight, kOutWidth, kDepthIn}, DT_FLOAT); in CreateSimpleMaxPoolGrad()
164 use_grad_grad ? kWidth : kOutWidth, kDepthIn}, in CreateSimpleMaxPoolGrad()
2468 use_grad_grad ? kOutWidth : kWidth, kDepthIn}, in TEST_F()
|