Home
last modified time | relevance | path

Searched refs:kParallel (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph.cc43 case CallContext::kParallel: in CallContextToString()
69 return CallContext::kParallel; in GetInstructionCallContext()
111 context == CallContext::kParallel); in AddCallSiteForInstruction()
216 if (callsite.context() == CallContext::kParallel) { in SetCallContexts()
217 context_to_add = CallContext::kParallel; in SetCallContexts()
Dflatten_call_graph.cc77 if (call_site.context() == CallContext::kParallel) { in FlattenNode()
Dcall_graph.h36 kParallel, enumerator
Dcall_graph_test.cc162 EXPECT_EQ(CallContext::kParallel, map_node.context()); in TEST_F()
239 EXPECT_EQ(CallContext::kParallel, map_callsite.context()); in TEST_F()
Dhlo_dataflow_analysis.cc476 if (call_graph_node.context() == CallContext::kParallel || in UpdateParameterValueSet()
797 call_graph_node.context() == CallContext::kParallel) { in InitializeInstructionValueSets()
Dhlo_rematerialization.cc960 if (callsite == nullptr || callsite->context() == CallContext::kParallel) { in CalledComputationsMemoryUsage()
Dcopy_insertion.cc1027 if (node.context() == CallContext::kParallel) { in AddSpecialCaseCopies()