Searched refs:tmp_axis (Results 1 – 1 of 1) sorted by relevance
57 int tmp_axis = wrap_around(axis, num_dimension); in compute_output_descriptor() local60 int split_size = input_descriptor.shape[tmp_axis] / num_splits; in compute_output_descriptor()63 output_descriptor.shape.set(tmp_axis, split_size); in compute_output_descriptor()64 coords.set(tmp_axis, idx * split_size); in compute_output_descriptor()71 split_size = input_descriptor.shape[tmp_axis]; in compute_output_descriptor()75 output_descriptor.shape.set(tmp_axis, split_size); in compute_output_descriptor()79 coords.set(tmp_axis, coord_value); in compute_output_descriptor()117 int tmp_axis = wrap_around(_axis, num_dimension); in configure_output() local119 …int split_size = (_size_splits.empty()) ? (input_descriptor.shape[tmp_axis] / _num_splits) : _size… in configure_output()122 split_size = input_descriptor.shape[tmp_axis]; in configure_output()[all …]