Home
last modified time | relevance | path

Searched refs:cast_node (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils_test.cc326 const NodeDef& cast_node = vectorized->node_def( in TEST() local
328 EXPECT_EQ(cast_node.input(0), vectorized->signature().input_arg(0).name()); in TEST()
330 strings::StrCat(cast_node.name(), ":y:0")); in TEST()
332 strings::StrCat(cast_node.name(), ":y:0")); in TEST()
407 const NodeDef& cast_node = vectorized->node_def( in TEST() local
409 EXPECT_EQ(cast_node.input(0), vectorized->signature().input_arg(0).name()); in TEST()
412 EXPECT_EQ(unpack_node.input(0), strings::StrCat(cast_node.name(), ":y:0")); in TEST()
485 const NodeDef& cast_node = map_defun_fn->node_def( in TEST() local
488 EXPECT_TRUE(cast_node.input(0) == control_input || in TEST()
489 cast_node.input(1) == control_input); in TEST()
[all …]
Drebatch.cc100 NodeDef cast_node; in AddCastNode() local
101 cast_node.set_op(kCastOp); in AddCastNode()
102 cast_node.add_input(input); in AddCastNode()
103 graph_utils::SetUniqueGraphNodeName(cast_node.op(), graph->graph(), in AddCastNode()
104 &cast_node); in AddCastNode()
105 AddNodeAttr("SrcT", src_t, &cast_node); in AddCastNode()
106 AddNodeAttr("DstT", dst_t, &cast_node); in AddCastNode()
108 return graph->AddNode(std::move(cast_node)); in AddCastNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc1227 const NodeDef* cast_node = nullptr; in TEST_F() local
1230 EXPECT_EQ(cast_node, nullptr); in TEST_F()
1231 cast_node = &node; in TEST_F()
1235 EXPECT_NE(cast_node, nullptr); in TEST_F()
1240 EXPECT_EQ(NodeName(node.input(0)), cast_node->name()); in TEST_F()
1277 const NodeDef* cast_node = nullptr; in TEST_F() local
1288 cast_node = &node; in TEST_F()
1291 EXPECT_NE(cast_node, nullptr); in TEST_F()
1296 EXPECT_EQ(NodeName(cast_node->input(0)), transpose_node->name()); in TEST_F()
1688 const NodeDef* cast_node = node_map.GetNode(optimized_cast_name); in TEST_F() local
[all …]