Searched refs:out_batches (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/ |
D | graph_functions_wrapper.c | 215 const uint32_t int_data_size, uint32_t* out_batches, uint32_t* out_height, in hexagon_controller_ExecuteGraph() argument 243 *out_batches = output.batches; in hexagon_controller_ExecuteGraph() 251 TFMLOGD("%d x %d x %d x %d, byte size = %d\n", *out_batches, *out_height, in hexagon_controller_ExecuteGraph() 259 uint32_t out_batches, out_height, out_width, out_depth; in hexagon_controller_ExecuteInceptionDummyData() local 268 &out_batches, &out_height, &out_width, &out_depth, in hexagon_controller_ExecuteInceptionDummyData() 273 s_output_values, out_batches * out_height * out_width * out_depth, in hexagon_controller_ExecuteInceptionDummyData() 275 TFMLOGD("%d x %d x %d x %d, size = %d\n", out_batches, out_height, in hexagon_controller_ExecuteInceptionDummyData() 279 out_batches * out_height * out_width * out_depth)); in hexagon_controller_ExecuteInceptionDummyData()
|
D | hexagon_controller.c | 184 const uint32_t out_batches = output0->batches; in hexagon_controller_ExecuteGraphWithBuffer() local 203 out_batches * out_height * out_width * out_depth, OUT_RANKING_SIZE, in hexagon_controller_ExecuteGraphWithBuffer() 205 TFMLOGD("%d x %d x %d x %d, byte size = %d, buf size = %d\n", out_batches, in hexagon_controller_ExecuteGraphWithBuffer()
|
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/include/ |
D | hexagon_controller.h | 110 const uint32_t int_data_size, uint32_t* out_batches, uint32_t* out_height,
|