| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| H A D | resolve_reorder_axes.cc | 29 // output_axes_order. Then adjust the shapes of the input and output arrays
33 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, argument
47 ShuffleArray(input_shape, input_axes_order, output_axes_order, output_shape,
73 reorder_op->output_axes_order,
77 reorder_op->output_axes_order,
|
| H A D | convert_reorder_axes.cc | 62 const AxesOrder& input_axes_order, const AxesOrder& output_axes_order) {
71 GetShuffleShape(input_axes_order, output_axes_order, &permutations_data);
120 const auto output_axes_order = reorder_op->output_axes_order; local
125 output_axes_order == AxesOrder::k1HWO) {
136 model, reorder_op, input_shape, input_axes_order, output_axes_order);
60 CreateTransposeFromReorderAxes( Model* model, ReorderAxesOperator* reorder_op, const Shape& input_shape, const AxesOrder& input_axes_order, const AxesOrder& output_axes_order) argument
|
| H A D | propagate_fixed_sizes.cc | 552 ShuffleDims(input_shape, op->input_axes_order, op->output_axes_order,
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| H A D | tooling_util.h | 275 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
284 AxesOrder output_axes_order, Shape* output_shape);
286 AxesOrder output_axes_order, const Shape& output_shape,
289 AxesOrder output_axes_order, const Shape& output_shape,
|
| H A D | tooling_util.cc | 1591 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, argument
1593 CHECK_EQ(AxesCount(input_axes_order), AxesCount(output_axes_order));
1598 if (input_axes_order == output_axes_order) {
1605 output_axes_order == AxesOrder::kHWIO) {
1613 output_axes_order == AxesOrder::kOHWI) {
1640 AxesOrder output_axes_order, Shape* output_shape) {
1642 output_axes_order == AxesOrder::k1HWO) {
1649 GetShuffleShape(input_axes_order, output_axes_order, &shuffle);
1660 AxesOrder output_axes_order,
1664 output_axes_order
1639 ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, AxesOrder output_axes_order, Shape* output_shape) argument
1659 ShuffleArrayTemplate(const Shape& input_shape, AxesOrder input_axes_order, AxesOrder output_axes_order, const Shape& output_shape, const T* input_data, T* output_data) argument
1733 ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order, AxesOrder output_axes_order, const Shape& output_shape, const uint8* input_data, uint8* output_data) argument
1740 ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order, AxesOrder output_axes_order, const Shape& output_shape, const float* input_data, float* output_data) argument
[all...] |
| H A D | export_tensorflow.cc | 109 const float* input_data, AxesOrder output_axes_order,
112 CHECK_EQ(AxesCount(output_axes_order), AxesCount(input_axes_order));
118 ShuffleDims(input_shape, input_axes_order, output_axes_order,
121 ShuffleArray(input_shape, input_axes_order, output_axes_order, shuffled_shape,
141 AxesOrder output_axes_order,
152 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order,
159 AxesOrder output_axes_order,
169 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order,
175 AxesOrder output_axes_order,
192 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order,
108 ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape, const float* input_data, AxesOrder output_axes_order, TensorProto* output_tensor, LegacyScalarPolicy legacy_scalar_policy) argument
138 ConvertFloatTensorConst(const string& name, const Shape& input_shape, const float* input_data, AxesOrder input_axes_order, AxesOrder output_axes_order, GraphDef* tensorflow_graph, LegacyScalarPolicy legacy_scalar_policy) argument
156 ConvertFloatTensorConst(const string& name, const Shape& input_shape, const float* input_data, AxesOrder input_axes_order, AxesOrder output_axes_order, GraphDef* tensorflow_graph) argument
173 ConvertFloatTensorConst(const Model& model, const string& name, AxesOrder input_axes_order, AxesOrder output_axes_order, GraphDef* tensorflow_graph) argument
[all...] |
| H A D | model.h | 625 AxesOrder output_axes_order; member in struct:toco::ReorderAxesOperator
|
| H A D | import_tensorflow.cc | 390 reorder->output_axes_order = AxesOrder::kOHWI;
443 reorder->output_axes_order = AxesOrder::k1HWO;
|