Home
last modified time | relevance | path

Searched refs:input_axes_order (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_reorder_axes.cc62 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()
123 const auto input_axes_order = reorder_op->input_axes_order; in Run() local
128 if (input_axes_order == AxesOrder::kHWIM && in Run()
140 model, reorder_op, input_shape, input_axes_order, output_axes_order); in Run()
Dresolve_reorder_axes.cc57 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes() argument
67 if (AxesCount(input_axes_order) == 2) { in ReorderAxes()
71 ShuffleArray(input_shape, input_axes_order, output_axes_order, output_shape, in ReorderAxes()
106 ReorderAxes<float, ArrayDataType::kFloat>(reorder_op->input_axes_order, in Run()
113 ReorderAxes<uint8, ArrayDataType::kUint8>(reorder_op->input_axes_order, in Run()
Dpropagate_fixed_sizes.cc650 ShuffleDims(input_shape, op->input_axes_order, op->output_axes_order, in ProcessReorderAxesOperator()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h280 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
288 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order,
290 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
293 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
Dtooling_util.cc2007 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape() argument
2009 CHECK_EQ(AxesCount(input_axes_order), AxesCount(output_axes_order)); in GetShuffleShape()
2014 if (input_axes_order == output_axes_order) { in GetShuffleShape()
2016 } else if (AxesCount(input_axes_order) == 2) { in GetShuffleShape()
2020 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2025 } else if (input_axes_order == AxesOrder::kHWIO && in GetShuffleShape()
2030 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2052 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleDims() argument
2054 if (input_axes_order == AxesOrder::kHWIM && in ShuffleDims()
2062 GetShuffleShape(input_axes_order, output_axes_order, &shuffle); in ShuffleDims()
[all …]
Dexport_tensorflow.cc120 void ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape, in ExportFloatArray() argument
124 CHECK_EQ(AxesCount(output_axes_order), AxesCount(input_axes_order)); in ExportFloatArray()
126 CHECK_EQ(input_shape.dimensions_count(), AxesCount(input_axes_order)); in ExportFloatArray()
130 ShuffleDims(input_shape, input_axes_order, output_axes_order, in ExportFloatArray()
133 ShuffleArray(input_shape, input_axes_order, output_axes_order, shuffled_shape, in ExportFloatArray()
152 AxesOrder input_axes_order, in ConvertFloatTensorConst() argument
164 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order, in ConvertFloatTensorConst()
170 AxesOrder input_axes_order, in ConvertFloatTensorConst() argument
173 ConvertFloatTensorConst(name, input_shape, input_data, input_axes_order, in ConvertFloatTensorConst()
179 AxesOrder input_axes_order, in ConvertFloatTensorConst() argument
[all …]
Dmodel.h851 AxesOrder input_axes_order;
Dimport_tensorflow.cc645 reorder->input_axes_order = AxesOrder::kHWIO; in ConvertConvOperator()
721 reorder->input_axes_order = AxesOrder::kHWIM; in ConvertDepthwiseConvOperator()