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
|