Home
last modified time | relevance | path

Searched refs:AxesOrder (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h276 int AxesCount(AxesOrder axes_order);
280 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
288 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order,
289 AxesOrder output_axes_order, Shape* output_shape);
290 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
291 AxesOrder output_axes_order, const Shape& output_shape,
293 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
294 AxesOrder output_axes_order, const Shape& output_shape,
Dtooling_util.cc2007 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape()
2020 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2021 output_axes_order == AxesOrder::kHWIO) { in GetShuffleShape()
2025 } else if (input_axes_order == AxesOrder::kHWIO && in GetShuffleShape()
2026 output_axes_order == AxesOrder::kOHWI) { in GetShuffleShape()
2030 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2031 output_axes_order == AxesOrder::kHWOI) { in GetShuffleShape()
2052 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleDims()
2053 AxesOrder output_axes_order, Shape* output_shape) { in ShuffleDims()
2054 if (input_axes_order == AxesOrder::kHWIM && in ShuffleDims()
[all …]
Dexport_tensorflow.cc120 void ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape, in ExportFloatArray()
121 const float* input_data, AxesOrder output_axes_order, in ExportFloatArray()
152 AxesOrder input_axes_order, in ConvertFloatTensorConst()
153 AxesOrder output_axes_order, in ConvertFloatTensorConst()
170 AxesOrder input_axes_order, in ConvertFloatTensorConst()
171 AxesOrder output_axes_order, in ConvertFloatTensorConst()
179 AxesOrder input_axes_order, in ConvertFloatTensorConst()
180 AxesOrder output_axes_order, in ConvertFloatTensorConst()
402 ConvertFloatTensorConst(model, weights_array_name, AxesOrder::kOHWI, in ConvertConvOperator()
403 AxesOrder::kHWIO, tensorflow_graph); in ConvertConvOperator()
[all …]
Dmodel.h180 enum class AxesOrder { enum
851 AxesOrder input_axes_order;
852 AxesOrder output_axes_order;
Dimport_tensorflow.cc645 reorder->input_axes_order = AxesOrder::kHWIO; in ConvertConvOperator()
646 reorder->output_axes_order = AxesOrder::kOHWI; in ConvertConvOperator()
721 reorder->input_axes_order = AxesOrder::kHWIM; in ConvertDepthwiseConvOperator()
722 reorder->output_axes_order = AxesOrder::k1HWO; in ConvertDepthwiseConvOperator()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_reorder_axes.cc62 const AxesOrder& input_axes_order, const AxesOrder& output_axes_order) { in CreateTransposeFromReorderAxes()
128 if (input_axes_order == AxesOrder::kHWIM && in Run()
129 output_axes_order == AxesOrder::k1HWO) { in Run()
Dresolve_reorder_axes.cc57 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes()