Searched refs:split_size (Results 1 – 4 of 4) sorted by relevance
40 int64_t split_size = 0; in SplitImpl() local43 split_size += in SplitImpl()46 TFLITE_DCHECK_EQ(split_size, input_dims->data[axis_value]); in SplitImpl()
42 int64_t split_size = output_dims->data[axis] * output_count; in SplitImpl() local44 TFLITE_DCHECK_EQ(split_size, input_dims->data[axis]); in SplitImpl()
612 size_t old_size, split_size; in k_mm_realloc() local665 split_size = old_size - new_size - MMLIST_HEAD_SIZE; in k_mm_realloc()672 split_b->buf_size = split_size | MM_BUFF_FREE | MM_BUFF_PREV_USED; in k_mm_realloc()705 split_size = MM_GET_BUF_SIZE(this_b) - new_size - MMLIST_HEAD_SIZE; in k_mm_realloc()711 split_b->buf_size = split_size | MM_BUFF_FREE | MM_BUFF_PREV_USED; in k_mm_realloc()
843 int64_t split_size = 0; in Split() local851 split_size += output_shapes[i]->Dims(axis); in Split()853 TFLITE_DCHECK_EQ(split_size, input_shape.Dims(axis)); in Split()
Completed in 10 milliseconds