Home
last modified time | relevance | path

Searched refs:transpose_op (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dreorder_reshape_transpose.cc110 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( in Run() local
113 if (transpose_op == nullptr) { in Run()
117 if (!OperatorReady(*model, transpose_op) || transpose_op->perm.empty()) { in Run()
123 auto reshape_it = FindOpWithOutput(*model, transpose_op->inputs[0]); in Run()
143 const string output_name = transpose_op->outputs[0]; in Run()
186 AvailableArrayName(*model, transpose_op->outputs[0] + "_exchange"); in Run()
188 new_intermediate_name, transpose_op->outputs[0]); in Run()
189 transpose_op->inputs[0] = input_name; in Run()
190 transpose_op->outputs[0] = new_intermediate_name; in Run()
205 transpose_op->inputs[0] = input_name; in Run()
[all …]
Dmerge_reshape_into_preceding_transpose.cc135 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( in Run() local
138 if (transpose_op == nullptr) { in Run()
142 if (!OperatorReady(*model, transpose_op) || transpose_op->perm.empty()) { in Run()
156 LogName(*transpose_op), LogName(*reshape_op)); in Run()
160 AddMessageF("Merging operations %s and %s", LogName(*transpose_op), in Run()
169 const auto& transpose_perm = transpose_op->perm; in Run()
180 model->GetArray(transpose_op->inputs[1]) in Run()
183 transpose_op->perm = merged_perm; in Run()
186 model->GetArray(transpose_op->outputs[0]).clear_shape(); in Run()
Dconvert_trivial_transpose_to_reshape.cc59 TransposeOperator* transpose_op = in Run() local
62 const auto& input_array = model->GetArray(transpose_op->inputs[0]); in Run()
63 const auto& output_array = model->GetArray(transpose_op->outputs[0]); in Run()
71 std::vector<int> const& perm = transpose_op->perm; in Run()
89 reshape_op->inputs.push_back(transpose_op->inputs[0]); in Run()
90 reshape_op->outputs = transpose_op->outputs; in Run()
93 string perm_array_name = transpose_op->inputs[1]; in Run()
112 CHECK_EQ(transpose_it->get(), transpose_op); in Run()
Dresolve_tensorflow_matmul.cc92 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); in Run() local
93 if (!transpose_op) { in Run()
99 transpose_op = new TransposeOperator; in Run()
100 transpose_op->inputs = { in Run()
105 transpose_op->outputs = { in Run()
107 model->GetOrCreateArray(transpose_op->outputs[0]); in Run()
108 model->operators.emplace(matmul_it, transpose_op); in Run()
110 DCHECK_EQ(transpose_op, FindTransposeOpWithInput(*model, input_rhs)); in Run()
118 input_rhs = transpose_op->outputs[0]; in Run()
Dconvert_reorder_axes.cc63 auto* transpose_op = new TransposeOperator; in CreateTransposeFromReorderAxes() local
66 transpose_op->inputs.push_back(reorder_op->inputs[0]); in CreateTransposeFromReorderAxes()
67 transpose_op->outputs = reorder_op->outputs; in CreateTransposeFromReorderAxes()
74 string perm_array_name = AvailableArrayName(*model, transpose_op->outputs[0]); in CreateTransposeFromReorderAxes()
75 transpose_op->inputs.push_back(perm_array_name); in CreateTransposeFromReorderAxes()
84 return transpose_op; in CreateTransposeFromReorderAxes()
139 auto* transpose_op = CreateTransposeFromReorderAxes( in Run() local
142 model->operators.emplace(reorder_it, transpose_op); in Run()
Dunroll_batch_matmul.cc190 auto* transpose_op = new TransposeOperator; in TransposeInput() local
191 transpose_op->inputs = {input, perm_array_name}; in TransposeInput()
192 transpose_op->outputs = {AvailableArrayName(*model, input + "/transpose")}; in TransposeInput()
193 auto& transpose_array = model->GetOrCreateArray(transpose_op->outputs[0]); in TransposeInput()
196 model->GetOrCreateArray(transpose_op->outputs[0]); in TransposeInput()
197 return transpose_op; in TransposeInput()
235 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); in Run() local
236 tail_it = model->operators.emplace(tail_it, transpose_op) + 1; in Run()
237 input_lhs = transpose_op->outputs[0]; in Run()
243 TransposeOperator* transpose_op = TransposeInput(input_rhs, model); in Run() local
[all …]
/external/tensorflow/tensorflow/python/ops/
Dtranspose_benchmark.py53 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/
Dexport_tensorflow.cc630 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
631 transpose_op->set_op("Transpose"); in ConvertFullyConnectedOperator()
632 transpose_op->set_name(transpose_output); in ConvertFullyConnectedOperator()
633 *transpose_op->add_input() = src_op.inputs[1]; in ConvertFullyConnectedOperator()
634 *transpose_op->add_input() = transpose_perm; in ConvertFullyConnectedOperator()
635 (*transpose_op->mutable_attr())["T"].set_type( in ConvertFullyConnectedOperator()
637 (*transpose_op->mutable_attr())["Tperm"].set_type(DT_INT32); in ConvertFullyConnectedOperator()
643 *matmul_op->add_input() = transpose_op->name(); in ConvertFullyConnectedOperator()
1282 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertTransposeOperator() local
1283 transpose_op->set_op("Transpose"); in ConvertTransposeOperator()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD102 "transpose_op.cc",
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc2290 const NodeDef* transpose_op = node->op() == "Conj" ? input : node; in TrySimplify() local
2293 if ((IsTranspose(*transpose_op) || IsConjugateTranspose(*transpose_op)) && in TrySimplify()
2295 NodeDef* new_op = AddCopyNode(optimized_node_name, transpose_op); in TrySimplify()
2298 new_op->set_op(transpose_op->op() == "Transpose" ? "ConjugateTranspose" in TrySimplify()
/external/tensorflow/tensorflow/core/kernels/
DBUILD908 ":transpose_op",
1167 name = "transpose_op",
1169 "transpose_op.cc",
1171 hdrs = ["transpose_op.h"],
5644 "transpose_op.h",
5838 "transpose_op.cc",
7089 hdrs = ["transpose_op.h"],
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_op_files.txt287 tensorflow/core/kernels/transpose_op.cc