Searched refs:index_to_axis (Results 1 – 2 of 2) sorted by relevance
131 std::vector<Axis> index_to_axis; in ExtractAxisFromIndex() local135 index_to_axis = {Axis::BATCH}; in ExtractAxisFromIndex()139 index_to_axis = {Axis::BATCH, Axis::CHANNELS}; in ExtractAxisFromIndex()143 index_to_axis = {Axis::BATCH, Axis::WIDTH, Axis::CHANNELS}; in ExtractAxisFromIndex()147 index_to_axis = {Axis::BATCH, Axis::HEIGHT, Axis::WIDTH, Axis::CHANNELS}; in ExtractAxisFromIndex()152 *axis = index_to_axis[index]; in ExtractAxisFromIndex()
2156 std::vector<Axis> index_to_axis = {Axis::BATCH, Axis::WIDTH, in Parse() local2158 attr.perm.b = axis_to_index[index_to_axis[perm.data[0]]]; in Parse()2160 attr.perm.w = axis_to_index[index_to_axis[perm.data[1]]]; in Parse()2161 attr.perm.c = axis_to_index[index_to_axis[perm.data[2]]]; in Parse()2163 std::vector<Axis> index_to_axis = {Axis::BATCH, Axis::CHANNELS}; in Parse() local2164 attr.perm.b = axis_to_index[index_to_axis[perm.data[0]]]; in Parse()2167 attr.perm.c = axis_to_index[index_to_axis[perm.data[1]]]; in Parse()