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()
|