| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| convert_trivial_transpose_to_reshape.cc | 59 TransposeOperator* transpose_op = local
62 const auto& input_array = model->GetArray(transpose_op->inputs[0]);
63 const auto& output_array = model->GetArray(transpose_op->outputs[0]);
71 std::vector<int> const& perm = transpose_op->perm;
89 reshape_op->inputs.push_back(transpose_op->inputs[0]);
90 reshape_op->outputs = transpose_op->outputs;
93 string perm_array_name = transpose_op->inputs[1];
112 CHECK_EQ(transpose_it->get(), transpose_op);
|
| reorder_reshape_transpose.cc | 110 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( local
113 if (transpose_op == nullptr) {
117 if (!OperatorReady(*model, transpose_op) || transpose_op->perm.empty()) {
123 auto reshape_it = FindOpWithOutput(*model, transpose_op->inputs[0]);
143 const string output_name = transpose_op->outputs[0];
186 AvailableArrayName(*model, transpose_op->outputs[0] + "_exchange");
188 new_intermediate_name, transpose_op->outputs[0]);
189 transpose_op->inputs[0] = input_name;
190 transpose_op->outputs[0] = new_intermediate_name
[all...] |
| merge_reshape_into_preceding_transpose.cc | 135 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( local
138 if (transpose_op == nullptr) {
142 if (!OperatorReady(*model, transpose_op) || transpose_op->perm.empty()) {
156 LogName(*transpose_op), LogName(*reshape_op));
160 AddMessageF("Merging operations %s and %s", LogName(*transpose_op),
169 const auto& transpose_perm = transpose_op->perm;
179 // Update transpose_op's constant buffer to contain the new permutation.
180 model->GetArray(transpose_op->inputs[1])
183 transpose_op->perm = merged_perm
[all...] |
| resolve_tensorflow_matmul.cc | 92 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); local
93 if (!transpose_op) {
99 transpose_op = new TransposeOperator;
100 transpose_op->inputs = {
105 transpose_op->outputs = {
107 model->GetOrCreateArray(transpose_op->outputs[0]);
108 model->operators.emplace(matmul_it, transpose_op);
110 DCHECK_EQ(transpose_op, FindTransposeOpWithInput(*model, input_rhs));
118 input_rhs = transpose_op->outputs[0];
|
| convert_reorder_axes.cc | 63 auto* transpose_op = new TransposeOperator; local
66 transpose_op->inputs.push_back(reorder_op->inputs[0]);
67 transpose_op->outputs = reorder_op->outputs;
74 string perm_array_name = AvailableArrayName(*model, transpose_op->outputs[0]);
75 transpose_op->inputs.push_back(perm_array_name);
84 return transpose_op;
139 auto* transpose_op = CreateTransposeFromReorderAxes( local
142 model->operators.emplace(reorder_it, transpose_op);
|
| unroll_batch_matmul.cc | 190 auto* transpose_op = new TransposeOperator; local
191 transpose_op->inputs = {input, perm_array_name};
192 transpose_op->outputs = {AvailableArrayName(*model, input + "/transpose")};
193 auto& transpose_array = model->GetOrCreateArray(transpose_op->outputs[0]);
196 model->GetOrCreateArray(transpose_op->outputs[0]);
197 return transpose_op;
235 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); local
236 tail_it = model->operators.emplace(tail_it, transpose_op) + 1;
237 input_lhs = transpose_op->outputs[0];
243 TransposeOperator* transpose_op = TransposeInput(input_rhs, model) local
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| transpose_benchmark.py | 53 transpose_op = array_ops.transpose(t, perm)
54 outputs.append(transpose_op)
56 with ops.control_dependencies([transpose_op]):
57 transpose_op = array_ops.transpose(t, perm)
58 outputs.append(transpose_op)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| export_tensorflow.cc | 630 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); local
631 transpose_op->set_op("Transpose");
632 transpose_op->set_name(transpose_output);
633 *transpose_op->add_input() = src_op.inputs[1];
634 *transpose_op->add_input() = transpose_perm;
635 (*transpose_op->mutable_attr())["T"].set_type(
637 (*transpose_op->mutable_attr())["Tperm"].set_type(DT_INT32);
643 *matmul_op->add_input() = transpose_op->name();
1282 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| arithmetic_optimizer.cc | 2290 const NodeDef* transpose_op = node->op() == "Conj" ? input : node; variable
[all...] |
