Searched refs:reorder_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_reorder_axes.cc | 30 Model* model, ReorderAxesOperator* reorder_op, const Shape& input_shape) { in CreateReshapeFromReorderAxes() argument
34 reshape_op->inputs.push_back(reorder_op->inputs[0]); in CreateReshapeFromReorderAxes()
35 reshape_op->outputs = reorder_op->outputs; in CreateReshapeFromReorderAxes()
61 Model* model, ReorderAxesOperator* reorder_op, const Shape& input_shape, in CreateTransposeFromReorderAxes() argument
66 transpose_op->inputs.push_back(reorder_op->inputs[0]); in CreateTransposeFromReorderAxes()
67 transpose_op->outputs = reorder_op->outputs; in CreateTransposeFromReorderAxes()
97 auto* reorder_op = static_cast<ReorderAxesOperator*>(reorder_it->get()); in Run() local
98 CHECK_EQ(reorder_op->inputs.size(), 1); in Run()
99 CHECK_EQ(reorder_op->outputs.size(), 1); in Run()
101 const auto& input_array_name = reorder_op->inputs[0]; in Run()
[all …]
|
| D | resolve_reorder_axes.cc | 89 auto* reorder_op = static_cast<ReorderAxesOperator*>(op); in Run() local
92 const std::string input_array_name = reorder_op->inputs[0]; in Run()
93 const std::string output_array_name = reorder_op->outputs[0]; in Run()
106 ReorderAxes<float, ArrayDataType::kFloat>(reorder_op->input_axes_order, in Run()
107 reorder_op->output_axes_order, in Run()
113 ReorderAxes<uint8, ArrayDataType::kUint8>(reorder_op->input_axes_order, in Run()
114 reorder_op->output_axes_order, in Run()
|