Searched defs:output_axes_order (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_reorder_axes.cc | 62 const AxesOrder& input_axes_order, const AxesOrder& output_axes_order) { in CreateTransposeFromReorderAxes()
124 const auto output_axes_order = reorder_op->output_axes_order; in Run() local
|
| D | resolve_reorder_axes.cc | 57 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2007 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape()
2053 AxesOrder output_axes_order, Shape* output_shape) { in ShuffleDims()
2073 AxesOrder output_axes_order, in ShuffleArrayTemplate()
2147 AxesOrder output_axes_order, const Shape& output_shape, in ShuffleArray()
2154 AxesOrder output_axes_order, const Shape& output_shape, in ShuffleArray()
|
| D | export_tensorflow.cc | 121 const float* input_data, AxesOrder output_axes_order, in ExportFloatArray()
153 AxesOrder output_axes_order, in ConvertFloatTensorConst()
171 AxesOrder output_axes_order, in ConvertFloatTensorConst()
180 AxesOrder output_axes_order, in ConvertFloatTensorConst()
|