Searched refs:batch_output (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op_gpu.h | 360 auto batch_output = output + offset; in TopKKernel() local
367 mergeShards(thread_count, k, shared_entries, top_k_heap, batch_output, in TopKKernel()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 1748 outputs = nest.map_structure(lambda batch_output: [batch_output],
1753 lambda output, batch_output: output.append(batch_output),
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_test.cc | 1758 std::vector<int16_t> batch_output = { in TEST() local
1774 1073741824, -1, batch_output.data()); in TEST()
1791 batch_output.data(), 1, 1); in TEST()
|