Searched refs:output_last_axis (Results 1 – 1 of 1) sorted by relevance
120 auto output_last_axis = output_type.getShape().size() - 1; in buildRescaleOpConvOutput() local121 uint32_t output_channels = output_type.getShape()[output_last_axis]; in buildRescaleOpConvOutput()