Home
last modified time | relevance | path

Searched defs:input_axes_order (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dconvert_reorder_axes.cc62 const AxesOrder& input_axes_order, const AxesOrder& output_axes_order) { in CreateTransposeFromReorderAxes()
124 const auto input_axes_order = reorder_op->input_axes_order; in Run() local
H A Dresolve_reorder_axes.cc58 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc2053 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape()
2098 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleDims()
2118 void ShuffleArrayTemplate(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleArrayTemplate()
2192 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleArray()
2199 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleArray()
H A Dexport_tensorflow.cc128 void ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape, in ExportFloatArray()
160 AxesOrder input_axes_order, in ConvertFloatTensorConst()
178 AxesOrder input_axes_order, in ConvertFloatTensorConst()
187 AxesOrder input_axes_order, in ConvertFloatTensorConst()