HomeSort by relevance Sort by last modified time
    Searched refs:transpose_op (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
convert_trivial_transpose_to_reshape.cc 29 TransposeOperator* transpose_op = local
32 const auto& output_array = model->GetArray(transpose_op->outputs[0]);
56 reshape_op->inputs.push_back(transpose_op->inputs[0]);
57 reshape_op->outputs = transpose_op->outputs;
60 string perm_array_name = transpose_op->inputs[1];
79 CHECK_EQ(transpose_it->get(), transpose_op);
resolve_tensorflow_matmul.cc 39 auto* transpose_op = new TransposeOperator; local
40 transpose_op->inputs = {
46 transpose_op->outputs = {
48 model->GetOrCreateArray(transpose_op->outputs[0]);
49 model->operators.emplace(matmul_it, transpose_op);
61 string 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;
135 auto* transpose_op = CreateTransposeFromReorderAxes( local
138 model->operators.emplace(reorder_it, transpose_op);
  /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/contrib/lite/toco/
export_tensorflow.cc 524 auto transpose_op = tensorflow_graph->add_node(); local
525 transpose_op->set_op("Transpose");
526 transpose_op->set_name(transpose_output);
527 *transpose_op->add_input() = src_op.inputs[1];
528 *transpose_op->add_input() = transpose_perm;
529 (*transpose_op->mutable_attr())["T"].set_type(
531 (*transpose_op->mutable_attr())["Tperm"].set_type(DT_INT32);
537 *matmul_op->add_input() = transpose_op->name();
1069 auto* transpose_op = tensorflow_graph->add_node(); local
    [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
arithmetic_optimizer.cc 1035 const NodeDef* transpose_op = node->op() == "Conj" ? input : node; local
    [all...]

Completed in 197 milliseconds