Home
last modified time | relevance | path

Searched refs:NodeWith (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers_test.cc37 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 …]
Dbuild_xla_ops_pass_test.cc52 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 …]
Dincrease_dynamism_for_auto_jit_pass_test.cc37 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 …]
Dnode_matchers.h156 ::testing::Matcher<const Node*> NodeWith(
234 ::testing::Matcher<const Node*> NodeWith(Ts... args) { in NodeWith() function
236 return impl::NodeWith(array); in NodeWith()
Dnode_matchers.cc387 ::testing::Matcher<const Node*> impl::NodeWith( in NodeWith() function in tensorflow::testing::matchers::impl
520 return Out(NodeWith(ConstantValue(val))); in Const()