Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h274 int AxesCount(AxesOrder axes_order);
278 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
286 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order,
287 AxesOrder output_axes_order, Shape* output_shape);
288 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
289 AxesOrder output_axes_order, const Shape& output_shape,
291 void ShuffleArray(const Shape& input_shape, AxesOrder input_axes_order,
292 AxesOrder output_axes_order, const Shape& output_shape,
Dtooling_util.cc2051 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order, in GetShuffleShape()
2064 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2065 output_axes_order == AxesOrder::kHWIO) { in GetShuffleShape()
2069 } else if (input_axes_order == AxesOrder::kHWIO && in GetShuffleShape()
2070 output_axes_order == AxesOrder::kOHWI) { in GetShuffleShape()
2074 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()
2075 output_axes_order == AxesOrder::kHWOI) { in GetShuffleShape()
2096 void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, in ShuffleDims()
2097 AxesOrder output_axes_order, Shape* output_shape) { in ShuffleDims()
2098 if (input_axes_order == AxesOrder::kHWIM && in ShuffleDims()
[all …]
Dexport_tensorflow.cc123 void ExportFloatArray(AxesOrder input_axes_order, const Shape& input_shape, in ExportFloatArray()
124 const float* input_data, AxesOrder output_axes_order, in ExportFloatArray()
155 AxesOrder input_axes_order, in ConvertFloatTensorConst()
156 AxesOrder output_axes_order, in ConvertFloatTensorConst()
173 AxesOrder input_axes_order, in ConvertFloatTensorConst()
174 AxesOrder output_axes_order, in ConvertFloatTensorConst()
182 AxesOrder input_axes_order, in ConvertFloatTensorConst()
183 AxesOrder output_axes_order, in ConvertFloatTensorConst()
406 ConvertFloatTensorConst(model, weights_array_name, AxesOrder::kOHWI, in ConvertConvOperator()
407 AxesOrder::kHWIO, tensorflow_graph); in ConvertConvOperator()
[all …]
Dmodel.h194 enum class AxesOrder { enum
898 AxesOrder input_axes_order;
899 AxesOrder output_axes_order;
Dimport_tensorflow.cc857 reorder->input_axes_order = AxesOrder::kHWIO; in ConvertConvOperator()
858 reorder->output_axes_order = AxesOrder::kOHWI; in ConvertConvOperator()
939 reorder->input_axes_order = AxesOrder::kHWIM; in ConvertDepthwiseConvOperator()
940 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()
129 if (input_axes_order == AxesOrder::kHWIM && in Run()
130 output_axes_order == AxesOrder::k1HWO) { in Run()
Dresolve_reorder_axes.cc57 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes()