Searched refs:output_axes_order (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_reorder_axes.cc | 62 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 output_axes_order = reorder_op->output_axes_order; in Run() local
129 output_axes_order == AxesOrder::k1HWO) { in Run()
140 model, reorder_op, input_shape, input_axes_order, output_axes_order); in Run()
|
| D | resolve_reorder_axes.cc | 57 void ReorderAxes(AxesOrder input_axes_order, AxesOrder output_axes_order, in ReorderAxes() argument
71 ShuffleArray(input_shape, input_axes_order, output_axes_order, output_shape, in ReorderAxes()
107 reorder_op->output_axes_order, in Run()
114 reorder_op->output_axes_order, in Run()
|
| D | propagate_fixed_sizes.cc | 650 ShuffleDims(input_shape, op->input_axes_order, op->output_axes_order, in ProcessReorderAxesOperator()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 280 void GetShuffleShape(AxesOrder input_axes_order, AxesOrder output_axes_order,
289 AxesOrder output_axes_order, Shape* output_shape);
291 AxesOrder output_axes_order, const Shape& output_shape,
294 AxesOrder output_axes_order, const Shape& output_shape,
|
| D | tooling_util.cc | 2007 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()
2021 output_axes_order == AxesOrder::kHWIO) { in GetShuffleShape()
2026 output_axes_order == AxesOrder::kOHWI) { in GetShuffleShape()
2031 output_axes_order == AxesOrder::kHWOI) { in GetShuffleShape()
2053 AxesOrder output_axes_order, Shape* output_shape) { in ShuffleDims() argument
2055 output_axes_order == AxesOrder::k1HWO) { in ShuffleDims()
2062 GetShuffleShape(input_axes_order, output_axes_order, &shuffle); in ShuffleDims()
2073 AxesOrder output_axes_order, in ShuffleArrayTemplate() argument
[all …]
|
| D | export_tensorflow.cc | 121 const float* input_data, AxesOrder output_axes_order, in ExportFloatArray() argument
124 CHECK_EQ(AxesCount(output_axes_order), 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()
153 AxesOrder output_axes_order, in ConvertFloatTensorConst() argument
164 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order, in ConvertFloatTensorConst()
171 AxesOrder output_axes_order, in ConvertFloatTensorConst() argument
174 output_axes_order, tensorflow_graph, in ConvertFloatTensorConst()
180 AxesOrder output_axes_order, in ConvertFloatTensorConst() argument
193 output_axes_order, tensorflow_graph); in ConvertFloatTensorConst()
|
| D | model.h | 852 AxesOrder output_axes_order;
|
| D | import_tensorflow.cc | 646 reorder->output_axes_order = AxesOrder::kOHWI; in ConvertConvOperator()
722 reorder->output_axes_order = AxesOrder::k1HWO; in ConvertDepthwiseConvOperator()
|