Searched refs:output_axes_order (Results 1 – 8 of 8) 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() argument
71 GetShuffleShape(input_axes_order, output_axes_order, &permutations_data); in CreateTransposeFromReorderAxes()
125 const auto output_axes_order = reorder_op->output_axes_order; in Run() local
130 output_axes_order == AxesOrder::k1HWO) { in Run()
140 model, reorder_op, input_shape, input_axes_order, output_axes_order); in Run()
|
| D | resolve_reorder_axes.cc | 57 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes() argument
71 ShuffleArray(input_shape, input_axes_order, output_axes_order, output_shape, in ReorderAxes()
107 reorder_op->output_axes_order, in Run()
114 reorder_op->output_axes_order, in Run()
|
| D | propagate_fixed_sizes.cc | 651 ShuffleDims(input_shape, op->input_axes_order, op->output_axes_order, in ProcessReorderAxesOperator()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 278 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
287 AxesOrder output_axes_order, Shape* output_shape);
289 AxesOrder output_axes_order, const Shape& output_shape,
292 AxesOrder output_axes_order, const Shape& output_shape,
|
| D | tooling_util.cc | 2051 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape() argument
2053 CHECK_EQ(AxesCount(input_axes_order), AxesCount(output_axes_order)); in GetShuffleShape()
2058 if (input_axes_order == output_axes_order) { in GetShuffleShape()
2065 output_axes_order == AxesOrder::kHWIO) { in GetShuffleShape()
2070 output_axes_order == AxesOrder::kOHWI) { in GetShuffleShape()
2075 output_axes_order == AxesOrder::kHWOI) { in GetShuffleShape()
2097 AxesOrder output_axes_order, Shape* output_shape) { in ShuffleDims() argument
2099 output_axes_order == AxesOrder::k1HWO) { in ShuffleDims()
2106 GetShuffleShape(input_axes_order, output_axes_order, &shuffle); in ShuffleDims()
2117 AxesOrder output_axes_order, in ShuffleArrayTemplate() argument
[all …]
|
| D | export_tensorflow.cc | 124 const float* input_data, AxesOrder output_axes_order, in ExportFloatArray() argument
127 CHECK_EQ(AxesCount(output_axes_order), AxesCount(input_axes_order)); in ExportFloatArray()
133 ShuffleDims(input_shape, input_axes_order, output_axes_order, in ExportFloatArray()
136 ShuffleArray(input_shape, input_axes_order, output_axes_order, shuffled_shape, in ExportFloatArray()
156 AxesOrder output_axes_order, in ConvertFloatTensorConst() argument
167 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order, in ConvertFloatTensorConst()
174 AxesOrder output_axes_order, in ConvertFloatTensorConst() argument
177 output_axes_order, tensorflow_graph, in ConvertFloatTensorConst()
183 AxesOrder output_axes_order, in ConvertFloatTensorConst() argument
196 output_axes_order, tensorflow_graph); in ConvertFloatTensorConst()
|
| D | model.h | 899 AxesOrder output_axes_order;
|
| D | import_tensorflow.cc | 858 reorder->output_axes_order = AxesOrder::kOHWI; in ConvertConvOperator()
940 reorder->output_axes_order = AxesOrder::k1HWO; in ConvertDepthwiseConvOperator()
|