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()
124 const auto input_axes_order = reorder_op->input_axes_order; in Run() local
129 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.cc651 ShuffleDims(input_shape, op->input_axes_order, op->output_axes_order, in ProcessReorderAxesOperator()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h278 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
286 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order,
288 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
291 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
Dtooling_util.cc2051 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()
2060 } else if (AxesCount(input_axes_order) == 2) { in GetShuffleShape()
2064 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2069 } else if (input_axes_order == AxesOrder::kHWIO && in GetShuffleShape()
2074 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2096 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleDims() argument
2098 if (input_axes_order == AxesOrder::kHWIM && in ShuffleDims()
2106 GetShuffleShape(input_axes_order, output_axes_order, &shuffle); in ShuffleDims()
[all …]
Dexport_tensorflow.cc123 void ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape, in ExportFloatArray() argument
127 CHECK_EQ(AxesCount(output_axes_order), AxesCount(input_axes_order)); in ExportFloatArray()
129 CHECK_EQ(input_shape.dimensions_count(), 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()
155 AxesOrder input_axes_order, in ConvertFloatTensorConst() argument
167 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order, in ConvertFloatTensorConst()
173 AxesOrder input_axes_order, in ConvertFloatTensorConst() argument
176 ConvertFloatTensorConst(name, input_shape, input_data, input_axes_order, in ConvertFloatTensorConst()
182 AxesOrder input_axes_order, in ConvertFloatTensorConst() argument
[all …]
Dmodel.h898 AxesOrder input_axes_order;
Dimport_tensorflow.cc857 reorder->input_axes_order = AxesOrder::kHWIO; in ConvertConvOperator()
939 reorder->input_axes_order = AxesOrder::kHWIM; in ConvertDepthwiseConvOperator()