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 …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc1205 const NodeDef* cast_node = nullptr; in TEST_F() local
1208 EXPECT_EQ(cast_node, nullptr); in TEST_F()
1209 cast_node = &node; in TEST_F()
1214 ASSERT_NE(cast_node, nullptr); in TEST_F()
1220 EXPECT_EQ(cast_node->name(), NodeName(node.input(0))); in TEST_F()
1257 const NodeDef* cast_node = nullptr; in TEST_F() local
1268 cast_node = &node; in TEST_F()
1271 ASSERT_NE(cast_node, nullptr); in TEST_F()
1278 ASSERT_EQ(cast_node->input_size(), 1); in TEST_F()
1279 EXPECT_EQ(NodeName(cast_node->input(0)), transpose_node->name()); in TEST_F()
[all …]
Darithmetic_optimizer.cc3636 NodeDef* cast_node = nullptr; in TrySimplify() local
3638 GetInputNode(reduction_node->input(input_index), &cast_node)); in TrySimplify()
3640 if (IsCastFromSupportedType(*cast_node, &original_index_type)) { in TrySimplify()
3641 reduction_node->set_input(input_index, cast_node->input(0)); in TrySimplify()
3644 cast_node->input(0)); in TrySimplify()