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()
125 const auto output_axes_order = reorder_op->output_axes_order; in Run() local
|
| D | resolve_reorder_axes.cc | 58 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2053 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape()
2099 AxesOrder output_axes_order, Shape* output_shape) { in ShuffleDims()
2119 AxesOrder output_axes_order, in ShuffleArrayTemplate()
2193 AxesOrder output_axes_order, const Shape& output_shape, in ShuffleArray()
2200 AxesOrder output_axes_order, const Shape& output_shape, in ShuffleArray()
|
| D | export_tensorflow.cc | 129 const float* input_data, AxesOrder output_axes_order, in ExportFloatArray()
161 AxesOrder output_axes_order, in ConvertFloatTensorConst()
179 AxesOrder output_axes_order, in ConvertFloatTensorConst()
188 AxesOrder output_axes_order, in ConvertFloatTensorConst()
|