Searched refs:NodeWith (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | node_matchers_test.cc | 37 using testing::matchers::NodeWith; 54 EXPECT_THAT(placeholder.node(), NodeWith(Op("Placeholder"))); in TEST() 55 EXPECT_THAT(placeholder.node(), NodeWith(Name("placeholder"))); in TEST() 57 NodeWith(Op("Placeholder"), Name("placeholder"))); in TEST() 59 NodeWith(Name("placeholder"), Op("Placeholder"))); in TEST() 60 EXPECT_THAT(placeholder.node(), NodeWith(Inputs())); in TEST() 62 NodeWith(Op("Placeholder"), Name("placeholder"), Inputs())); in TEST() 64 EXPECT_EQ(Explain(placeholder.node(), NodeWith(Op("Add"))), in TEST() 66 EXPECT_EQ(Explain(placeholder.node(), NodeWith(Name("add"))), in TEST() 68 EXPECT_EQ(Explain(placeholder.node(), NodeWith(Inputs(Out(NodeWith())))), in TEST() [all …]
|
D | build_xla_ops_pass_test.cc | 52 using ::tensorflow::testing::matchers::NodeWith; 150 EXPECT_THAT(write_op_new, NodeWith(CtrlDeps(NodeWith(Op("_XlaRun"))))); in TEST_F() 186 auto xla_compile = NodeWith(Op("_XlaCompile"), Attr("must_compile", false)); in TEST_F() 188 NodeWith(Op("Switch"), Inputs(Out(0, xla_compile), Out(1, xla_compile))); in TEST_F() 190 NodeWith(Op("_XlaRun"), Inputs(Out(1, predicated_compilation_key))); in TEST_F() 192 NodeWith(Op("PartitionedCall"), in TEST_F() 193 CtrlDeps(NodeWith(Op("Identity"), in TEST_F() 195 auto merge = NodeWith(Op("Merge"), Inputs(Out(tf_call), Out(xla_run))); in TEST_F() 196 auto assign_var = NodeWith(Op("AssignVariableOp"), Inputs(_, Out(merge))); in TEST_F() 225 NodeWith(Op("_XlaRun"), Inputs(Out(NodeWith(Op("_XlaCompile"))))); in TEST_F() [all …]
|
D | increase_dynamism_for_auto_jit_pass_test.cc | 37 using testing::matchers::NodeWith; 118 auto m_input = Out(NodeWith(Op("Placeholder"), Name("input"))); in TEST() 119 auto m_begin_s64 = Out(NodeWith( in TEST() 120 Op("Cast"), Inputs(Out(NodeWith(Op("Placeholder"), Name("begin")))))); in TEST() 121 auto m_input_shape = Out(NodeWith(Op("Shape"), Inputs(m_input))); in TEST() 122 auto m_slice_size_0 = Out(NodeWith( in TEST() 125 Out(NodeWith(Op("Slice"), AssignedDevice(kHostName), in TEST() 127 Out(NodeWith(Op("Slice"), AssignedDevice(kHostName), in TEST() 129 auto m_dynamic_slice_size = Out(NodeWith( in TEST() 135 auto m_dynamic_slice = NodeWith( in TEST() [all …]
|
D | node_matchers.h | 156 ::testing::Matcher<const Node*> NodeWith( 234 ::testing::Matcher<const Node*> NodeWith(Ts... args) { in NodeWith() function 236 return impl::NodeWith(array); in NodeWith()
|
D | node_matchers.cc | 387 ::testing::Matcher<const Node*> impl::NodeWith( in NodeWith() function in tensorflow::testing::matchers::impl 520 return Out(NodeWith(ConstantValue(val))); in Const()
|