| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_reshape_attributes.cc | 30 auto* reshape_op = reshape_it->get(); local
31 if (reshape_op->type != OperatorType::kTensorFlowReshape) {
35 auto* op = static_cast<TensorFlowReshapeOperator*>(reshape_op);
39 if (IsConstantParameterArray(*model, reshape_op->inputs[1])) {
40 const auto& constant_input_array = model->GetArray(reshape_op->inputs[1]);
|
| convert_trivial_stack_to_reshape.cc | 54 auto* reshape_op = new TensorFlowReshapeOperator; local
55 reshape_op->inputs = {stack_op->inputs[0]};
56 reshape_op->outputs = stack_op->outputs;
64 reshape_op->inputs.push_back(shape_array_name);
73 const auto reshape_it = model->operators.emplace(stack_it, reshape_op);
|
| convert_trivial_transpose_to_reshape.cc | 53 auto* reshape_op = new TensorFlowReshapeOperator; local
56 reshape_op->inputs.push_back(transpose_op->inputs[0]);
57 reshape_op->outputs = transpose_op->outputs;
65 reshape_op->inputs.push_back(shape_array_name);
77 const auto reshape_it = model->operators.emplace(transpose_it, reshape_op);
|
| remove_trivial_reshape.cc | 77 auto* reshape_op = reshape_it->get(); local
78 if (reshape_op->type != OperatorType::kTensorFlowReshape) {
82 if (!IsReshapeTrivial(*model, *reshape_op, this)) {
86 AddMessageF("Removing trivial %s", LogName(*reshape_op));
88 CHECK_EQ(reshape_op->inputs.size(), 2);
|
| convert_expanddims_to_reshape.cc | 68 auto* reshape_op = new TensorFlowReshapeOperator; local
71 reshape_op->inputs.push_back(expand_op->inputs[0]);
72 reshape_op->outputs = expand_op->outputs;
80 reshape_op->inputs.push_back(shape_array_name);
93 const auto reshape_it = model->operators.emplace(expand_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;
129 auto* reshape_op = local
131 const auto reshape_it = model->operators.emplace(reorder_it, reshape_op);
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| export_tensorflow.cc | 502 auto* reshape_op = tensorflow_graph->add_node(); local
503 reshape_op->set_op("Reshape");
504 reshape_op->set_name(reshape_output);
505 reshape_op->add_input(src_op.inputs[0]);
506 reshape_op->add_input(reshape_shape);
507 (*reshape_op->mutable_attr())["T"].set_type(
692 auto* reshape_op = tensorflow_graph->add_node(); local
693 reshape_op->set_op("Reshape");
694 reshape_op->set_name(reshape_output);
695 *reshape_op->add_input() = src_op.inputs[0]
733 auto* reshape_op = tensorflow_graph->add_node(); local
908 auto* reshape_op = tensorflow_graph->add_node(); 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...] |
