Searched refs:arg_feed_y (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/tools/ |
| D | aot_compiled_test.cc | 47 Eigen::Tensor<float, 2, Eigen::RowMajor> arg_feed_y(5000, 4000); in TEST() local
49 arg_feed_y.setRandom(); in TEST()
56 arg_feed_x.contract(arg_feed_y, product_dims); in TEST()
59 model.set_arg_feed_y_data(arg_feed_y.data()); in TEST()
81 Eigen::Tensor<float, 2, Eigen::RowMajor> arg_feed_y(5000, 4000); in TEST() local
83 arg_feed_y.setRandom(); in TEST()
90 arg_feed_x.contract(arg_feed_y, product_dims); in TEST()
93 model.set_arg_feed_y_data(arg_feed_y.data()); in TEST()
110 Eigen::Tensor<float, 2, Eigen::RowMajor> arg_feed_y(5, 4); in TEST() local
112 arg_feed_y.setRandom(); in TEST()
[all …]
|