| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_reshape_attributes.cc | 33 auto* reshape_op = reshape_it->get(); local
34 if (reshape_op->type != OperatorType::kReshape) {
38 auto* op = static_cast<TensorFlowReshapeOperator*>(reshape_op);
42 if (IsConstantParameterArray(*model, reshape_op->inputs[1])) {
43 const auto& constant_input_array = model->GetArray(reshape_op->inputs[1]);
|
| convert_squeeze_to_reshape.cc | 68 auto* reshape_op = new TensorFlowReshapeOperator; local
69 reshape_op->inputs = {
74 reshape_op->outputs = squeeze_op->outputs;
77 LogName(*reshape_op));
80 const auto reshape_it = model->operators.emplace(squeeze_it, reshape_op);
|
| remove_trivial_reshape.cc | 89 auto* reshape_op = reshape_it->get(); local
90 if (reshape_op->type != OperatorType::kReshape) {
94 if (!IsReshapeTrivial(*model, *reshape_op, this)) {
95 AddMessageF("%s is not trivial", LogName(*reshape_op));
99 AddMessageF("Removing trivial %s", LogName(*reshape_op));
101 CHECK_EQ(reshape_op->inputs.size(), 2);
|
| reorder_reshape_transpose.cc | 128 TensorFlowReshapeOperator* reshape_op = local
131 if (reshape_op == nullptr) {
136 if (!OperatorReady(*model, reshape_op) || reshape_op->shape.empty()) {
141 const string input_name = reshape_op->inputs[0];
142 const string intermediate_name = reshape_op->outputs[0];
177 if (!ReshapeIsEquivalentToTranspose(*model, reshape_op,
191 reshape_op->inputs[0] = new_intermediate_name;
192 reshape_op->outputs[0] = output_name;
206 reshape_op->inputs[0] = output_name
[all...] |
| merge_reshape_into_preceding_transpose.cc | 109 auto* reshape_op = ConvertOperator<TensorFlowReshapeOperator*>( local
112 if (reshape_op == nullptr) {
116 if (!OperatorReady(*model, reshape_op) || reshape_op->shape.empty()) {
120 const string intermediate_name = reshape_op->inputs[0];
121 const string output_name = reshape_op->outputs[0];
146 if (!ReshapeIsEquivalentToTranspose(*model, reshape_op,
156 LogName(*transpose_op), LogName(*reshape_op));
161 LogName(*reshape_op));
166 auto merged_perm = ReshapeToTranspose(*model, reshape_op);
[all...] |
| move_binary_operator_before_reshape.cc | 121 Operator* reshape_op = reshape_it->get(); local
122 if (reshape_op->type != OperatorType::kReshape) {
124 LogName(*binary_op), LogName(*reshape_op));
127 const auto& reshape_input_array = model->GetArray(reshape_op->inputs[0]);
137 model->GetArray(reshape_op->outputs[0]).shape())) {
148 LogName(*binary_op), LogName(*reshape_op));
159 LogName(*binary_op), LogName(*reshape_op));
165 LogName(*reshape_op));
168 std::iter_swap(reshape_op->inputs.begin(),
170 std::iter_swap(reshape_op->outputs.begin(), binary_op->outputs.begin())
[all...] |
| convert_trivial_pack_to_reshape.cc | 57 auto* reshape_op = new TensorFlowReshapeOperator; local
58 reshape_op->inputs = {pack_op->inputs[0]};
59 reshape_op->outputs = pack_op->outputs;
67 reshape_op->inputs.push_back(shape_array_name);
76 const auto reshape_it = model->operators.emplace(pack_it, reshape_op);
|
| convert_reorder_axes.cc | 31 auto* reshape_op = new TensorFlowReshapeOperator; local
34 reshape_op->inputs.push_back(reorder_op->inputs[0]);
35 reshape_op->outputs = reorder_op->outputs;
45 AvailableArrayName(*model, reshape_op->outputs[0]);
46 reshape_op->inputs.push_back(reshape_array_name);
56 return reshape_op;
133 auto* reshape_op = local
135 const auto reshape_it = model->operators.emplace(reorder_it, reshape_op);
|
| convert_expanddims_to_reshape.cc | 66 auto* reshape_op = new TensorFlowReshapeOperator; local
69 reshape_op->inputs.push_back(expand_op->inputs[0]);
70 reshape_op->outputs = expand_op->outputs;
78 reshape_op->inputs.push_back(shape_array_name);
91 const auto reshape_it = model->operators.emplace(expand_it, reshape_op);
|
| convert_trivial_transpose_to_reshape.cc | 86 auto* reshape_op = new TensorFlowReshapeOperator; local
89 reshape_op->inputs.push_back(transpose_op->inputs[0]);
90 reshape_op->outputs = transpose_op->outputs;
98 reshape_op->inputs.push_back(shape_array_name);
110 const auto reshape_it = model->operators.emplace(transpose_it, reshape_op);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| export_tensorflow.cc | 608 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); local
609 reshape_op->set_op("Reshape");
610 reshape_op->set_name(reshape_output);
611 reshape_op->add_input(src_op.inputs[0]);
612 reshape_op->add_input(reshape_shape);
613 (*reshape_op->mutable_attr())["T"].set_type(
822 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); local
863 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); local
1044 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| eigen_spatial_convolutions_test.cc | 1494 const auto reshape_op = local
1661 const auto reshape_op = local
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes_test.cc | 479 Output reshape_op = local
480 Reshape(root.WithOpName("reshape_op"), constant_op, {10, 2});
485 TestQuantizedVersusFloatGraph(float_graph_def, {}, {"reshape_op"});
[all...] |
