Home
last modified time | relevance | path

Searched defs:index_to_axis (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_helper.cc131 std::vector<Axis> index_to_axis; in ExtractAxisFromIndex() local
Dmodel_builder.cc2156 std::vector<Axis> index_to_axis = {Axis::BATCH, Axis::WIDTH, in Parse() local
2163 std::vector<Axis> index_to_axis = {Axis::BATCH, Axis::CHANNELS}; in Parse() local