OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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