Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/toco/
Dtooling_util.h271 int AxesCount(AxesOrder axes_order);
275 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
283 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order,
284 AxesOrder output_axes_order, Shape* output_shape);
285 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
286 AxesOrder output_axes_order, const Shape& output_shape,
288 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
289 AxesOrder output_axes_order, const Shape& output_shape,
Dtooling_util.cc1591 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape()
1604 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
1605 output_axes_order == AxesOrder::kHWIO) { in GetShuffleShape()
1612 } else if (input_axes_order == AxesOrder::kHWIO && in GetShuffleShape()
1613 output_axes_order == AxesOrder::kOHWI) { in GetShuffleShape()
1639 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleDims()
1640 AxesOrder output_axes_order, Shape* output_shape) { in ShuffleDims()
1641 if (input_axes_order == AxesOrder::kHWIM && in ShuffleDims()
1642 output_axes_order == AxesOrder::k1HWO) { in ShuffleDims()
1659 void ShuffleArrayTemplate(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleArrayTemplate()
[all …]
Dexport_tensorflow.cc108 void ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape, in ExportFloatArray()
109 const float* input_data, AxesOrder output_axes_order, in ExportFloatArray()
140 AxesOrder input_axes_order, in ConvertFloatTensorConst()
141 AxesOrder output_axes_order, in ConvertFloatTensorConst()
158 AxesOrder input_axes_order, in ConvertFloatTensorConst()
159 AxesOrder output_axes_order, in ConvertFloatTensorConst()
174 AxesOrder input_axes_order, in ConvertFloatTensorConst()
175 AxesOrder output_axes_order, in ConvertFloatTensorConst()
347 ConvertFloatTensorConst(model, weights_array_name, AxesOrder::kOHWI, in ConvertConvOperator()
348 AxesOrder::kHWIO, tensorflow_graph); in ConvertConvOperator()
[all …]
Dmodel.h135 enum class AxesOrder { enum
624 AxesOrder input_axes_order;
625 AxesOrder output_axes_order;
Dimport_tensorflow.cc389 reorder->input_axes_order = AxesOrder::kHWIO; in ConvertConvOperator()
390 reorder->output_axes_order = AxesOrder::kOHWI; in ConvertConvOperator()
442 reorder->input_axes_order = AxesOrder::kHWIM; in ConvertDepthwiseConvOperator()
443 reorder->output_axes_order = AxesOrder::k1HWO; in ConvertDepthwiseConvOperator()
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dconvert_reorder_axes.cc62 const AxesOrder& input_axes_order, const AxesOrder& output_axes_order) { in CreateTransposeFromReorderAxes()
124 if (input_axes_order == AxesOrder::kHWIM && in Run()
125 output_axes_order == AxesOrder::k1HWO) { in Run()
Dresolve_reorder_axes.cc33 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes()