Searched defs:output_elems (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 711 std::vector<T> output_elems(2048, static_cast<T>(18)); in RunTest() local
775 std::vector<T> output_elems(2048, static_cast<T>(18)); in RunTest() local
846 std::vector<T> output_elems(2048 * 256, static_cast<T>(18)); in RunTest() local
915 std::vector<T> output_elems(2048 * 256, static_cast<T>(18)); in RunTest() local
1059 std::vector<T> output_elems(640, static_cast<T>(18)); in RunTest() local
1125 std::vector<T> output_elems(640, static_cast<T>(18)); in RunTest() local
1195 std::vector<T> output_elems(640, static_cast<T>(18)); in RunTest() local
1263 std::vector<T> output_elems(4096, static_cast<T>(18)); in RunTest() local
1393 std::vector<T> output_elems(512, static_cast<T>(1024)); in RunTest() local
1457 std::vector<T> output_elems(8, static_cast<T>(1024)); in RunTest() local
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 325 jlong* output_elems = in Java_org_tensorflow_Graph_whileLoop() local
|