| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_reorder_axes.cc | 28 // Reorder the elements of an input_array according to the input_axes_order and
33 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order,
43 if (AxesCount(input_axes_order) == 2) {
47 ShuffleArray(input_shape, input_axes_order, output_axes_order, output_shape,
72 ReorderAxes<float, ArrayDataType::kFloat>(reorder_op->input_axes_order,
76 ReorderAxes<uint8, ArrayDataType::kUint8>(reorder_op->input_axes_order,
|
| 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);
119 const auto input_axes_order = reorder_op->input_axes_order; local
124 if (input_axes_order == AxesOrder::kHWIM &&
136 model, reorder_op, input_shape, input_axes_order, output_axes_order);
|
| propagate_fixed_sizes.cc | 552 ShuffleDims(input_shape, op->input_axes_order, op->output_axes_order,
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| tooling_util.h | 275 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
283 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order,
285 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
288 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
|
| tooling_util.cc | [all...] |
| export_tensorflow.cc | 108 void ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape,
112 CHECK_EQ(AxesCount(output_axes_order), AxesCount(input_axes_order));
114 CHECK_EQ(input_shape.dimensions_count(), 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,
140 AxesOrder input_axes_order,
152 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order,
158 AxesOrder input_axes_order,
169 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order,
174 AxesOrder input_axes_order,
[all...] |
| model.h | 624 AxesOrder input_axes_order; member in struct:toco::ReorderAxesOperator
[all...] |
| import_tensorflow.cc | 389 reorder->input_axes_order = AxesOrder::kHWIO;
442 reorder->input_axes_order = AxesOrder::kHWIM;
[all...] |
