Home
last modified time | relevance | path

Searched refs:kSequential (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph.cc41 case CallContext::kSequential: in CallContextToString()
60 return CallContext::kSequential; in GetInstructionCallContext()
110 CHECK(context == CallContext::kSequential || in AddCallSiteForInstruction()
200 node.set_context(CallContext::kSequential); in SetCallContexts()
219 CHECK_EQ(callsite.context(), CallContext::kSequential); in SetCallContexts()
358 if (node.context() == CallContext::kSequential && in IsFlattened()
Dcall_graph_test.cc111 EXPECT_EQ(CallContext::kSequential, node.context()); in TEST_F()
129 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
135 EXPECT_EQ(CallContext::kSequential, unreachable_node.context()); in TEST_F()
153 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
187 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
195 EXPECT_EQ(CallContext::kSequential, called_node.context()); in TEST_F()
232 EXPECT_EQ(CallContext::kSequential, call_callsite.context()); in TEST_F()
282 EXPECT_EQ(CallContext::kSequential, conditional_callsite.context()); in TEST_F()
365 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
Dflatten_call_graph.cc80 CHECK_EQ(call_site.context(), CallContext::kSequential); in FlattenNode()
100 CallContext::kSequential) { in FlattenNode()
Dhlo_liveness_analysis.cc201 if (call_graph_node.context() == CallContext::kSequential) { in PropagateLivenessToParameterCallers()
231 if (call_graph_node.context() == CallContext::kSequential) { in PropagateLivenessThroughControlFlow()
Dcall_graph.h41 kSequential, enumerator
Dhlo_dataflow_analysis.cc480 CHECK_EQ(call_graph_node.context(), CallContext::kSequential); in UpdateParameterValueSet()
692 if (call_graph_node.context() == CallContext::kSequential) { in Propagate()
Dhlo_rematerialization.cc1133 callsite->context() == CallContext::kSequential && in RematerializeComputation()
1240 if (node.context() == CallContext::kSequential) { in Run()
Dcopy_insertion.cc1030 TF_RET_CHECK(node.context() == CallContext::kSequential); in AddSpecialCaseCopies()
1221 if (node.context() == CallContext::kSequential && in IsWhileBody()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk.cc53 case Thunk::kSequential: in ThunkKindToString()
Dsequential_thunk.cc26 : Thunk(Kind::kSequential, hlo), thunks_(std::move(thunks)) {} in SequentialThunk()
Dthunk.h60 kSequential, enumerator