Searched refs:idx_in (Results 1 – 3 of 3) sorted by relevance
33 size_t idx_in = 0; in BatchToSpace() local52 …idx_in = (((idx_oh + crop_up) % block_num) * block_num + ((idx_ow + crop_lft) % block_num)) * on +… in BatchToSpace()53 input_pos = idx_in * ic; in BatchToSpace()55 input_pos = (input_pos + ((idx_oh + crop_up) - (idx_in / (on * block_num))) / block_num) * iw; in BatchToSpace()56 input_pos = (input_pos + ((idx_ow + crop_lft) - ((idx_in / on) % block_num)) / block_num); in BatchToSpace()
29 size_t idx_in = 0; in SpaceToBatch() local38 idx_in = pos / temp_stride; in SpaceToBatch()52 …n = (((idx_ih + pad_up) % block_num) * block_num + ((idx_iw + pad_lft) % block_num)) * in + idx_in; in SpaceToBatch()
642 int idx_in = 0, idx_out = 0; in copy_metadata() local644 parse_meta_type(inspec, &type_in, &idx_in, &istream_spec); in copy_metadata()693 SET_DICT(type_in, meta_in, ic, idx_in); in copy_metadata()