/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | reorder_reshape_transpose.cc | 110 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 …]
|
D | merge_reshape_into_preceding_transpose.cc | 135 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()
|
D | convert_trivial_transpose_to_reshape.cc | 59 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()
|
D | resolve_tensorflow_matmul.cc | 92 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()
|
D | convert_reorder_axes.cc | 63 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()
|
D | unroll_batch_matmul.cc | 190 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/ |
D | 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/ |
D | export_tensorflow.cc | 630 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/ |
D | BUILD | 102 "transpose_op.cc",
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | arithmetic_optimizer.cc | 2290 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/ |
D | BUILD | 908 ":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/ |
D | tf_op_files.txt | 287 tensorflow/core/kernels/transpose_op.cc
|