Searched refs:split_dimensions (Results 1 – 4 of 4) sorted by relevance
36 const int split_dimensions = input_dims->size; in SplitImpl() local37 int axis = axis_value < 0 ? axis_value + split_dimensions : axis_value; in SplitImpl()39 TFLITE_DCHECK_LT(axis, split_dimensions); in SplitImpl()40 TFLITE_DCHECK_EQ(output_dims->size, split_dimensions); in SplitImpl()51 for (int i = axis + 1; i < split_dimensions; ++i) { in SplitImpl()
35 const int split_dimensions = input_dims->size; in SplitImpl() local37 TFLITE_DCHECK_LT(axis_value, split_dimensions); in SplitImpl()38 TFLITE_DCHECK_EQ(output0->dims->size, split_dimensions); in SplitImpl()53 for (int i = axis_value + 1; i < split_dimensions; ++i) { in SplitImpl()
837 const int split_dimensions = input_shape.DimensionsCount(); in Split() local838 int axis = params.axis < 0 ? params.axis + split_dimensions : params.axis; in Split()840 TFLITE_DCHECK_LT(axis, split_dimensions); in Split()844 TFLITE_DCHECK_EQ(output_shapes[i]->DimensionsCount(), split_dimensions); in Split()845 for (int j = 0; j < split_dimensions; j++) { in Split()860 for (int i = axis + 1; i < split_dimensions; ++i) { in Split()
124 number of blocks along the `split_dimensions`, and the blocks are scattered