Searched refs:kSequential (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph.cc | 40 case CallContext::kSequential: in CallContextToString()
59 return CallContext::kSequential; in GetInstructionCallContext()
106 CHECK(context == CallContext::kSequential || in AddCallSiteForInstruction()
196 node.set_context(CallContext::kSequential); in SetCallContexts()
215 CHECK_EQ(callsite.context(), CallContext::kSequential); in SetCallContexts()
317 if (node.context() == CallContext::kSequential && in IsFlattened()
|
| D | call_graph_test.cc | 109 EXPECT_EQ(CallContext::kSequential, node.context()); in TEST_F()
126 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
131 EXPECT_EQ(CallContext::kSequential, unreachable_node.context()); in TEST_F()
148 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
181 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
189 EXPECT_EQ(CallContext::kSequential, called_node.context()); in TEST_F()
226 EXPECT_EQ(CallContext::kSequential, call_callsite.context()); in TEST_F()
274 EXPECT_EQ(CallContext::kSequential, conditional_callsite.context()); in TEST_F()
343 EXPECT_EQ(CallContext::kSequential, entry_node.context()); in TEST_F()
|
| D | flatten_call_graph.cc | 77 CHECK_EQ(call_site.context(), CallContext::kSequential); in FlattenNode()
97 CallContext::kSequential) { in FlattenNode()
|
| D | call_graph.h | 41 kSequential, enumerator
|
| D | hlo_dataflow_analysis.cc | 436 CHECK_EQ(call_graph_node.context(), CallContext::kSequential); in UpdateParameterValueSet()
645 if (call_graph_node.context() == CallContext::kSequential) { in Propagate()
|
| D | copy_insertion.cc | 958 TF_RET_CHECK(node.context() == CallContext::kSequential); in AddSpecialCaseCopies()
1138 if (node.context() == CallContext::kSequential && in IsWhileBody()
|
| D | hlo_rematerialization.cc | 1116 callsite->context() == CallContext::kSequential && in RematerializeComputation()
1228 if (node.context() == CallContext::kSequential) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | sequential_thunk.cc | 25 : Thunk(Kind::kSequential, hlo), thunks_(std::move(thunks)) {} in SequentialThunk()
|
| D | thunk.h | 54 kSequential, enumerator
|
| D | ir_emitter_unnested.cc | 1115 CHECK(LastThunk()->kind() == Thunk::Kind::kSequential); in EmitReductionToScalar()
1285 CHECK(LastThunk()->kind() == Thunk::Kind::kSequential); in EmitColumnReduction()
1555 CHECK(LastThunk()->kind() == Thunk::Kind::kSequential); in EmitRowReduction()
|