Home
last modified time | relevance | path

Searched refs:ShuffleDims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h286 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order,
Dtooling_util.cc2096 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleDims() function
Dexport_tensorflow.cc133 ShuffleDims(input_shape, input_axes_order, output_axes_order, in ExportFloatArray()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc651 ShuffleDims(input_shape, op->input_axes_order, op->output_axes_order, in ProcessReorderAxesOperator()