Searched refs:transpose_node (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer_test.cc736 const NodeDef* transpose_node = nullptr; local
739 EXPECT_EQ(transpose_node, nullptr);
741 transpose_node = &node;
744 EXPECT_NE(transpose_node, nullptr);
748 EXPECT_EQ(NodeName(node.input(0)), transpose_node->name());
1047 const NodeDef* transpose_node = local
1056 EXPECT_EQ(transpose_node->input(0), inputs_node->name());
1057 EXPECT_EQ(cast_node->input(0), transpose_node->name());
H A Darithmetic_optimizer.cc157 bool IsInnerMatrixTransposeNode(const NodeDef& transpose_node, argument
159 if (transpose_node.op() != "Transpose" &&
160 transpose_node.op() != "ConjugateTranspose") {
163 const NodeDef* perm_node = node_map->GetNode(transpose_node.input(1));

Completed in 68 milliseconds