Home
last modified time | relevance | path

Searched refs:CallContext (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph.cc37 string CallContextToString(CallContext context) { in CallContextToString()
39 case CallContext::kNone: in CallContextToString()
41 case CallContext::kSequential: in CallContextToString()
43 case CallContext::kParallel: in CallContextToString()
45 case CallContext::kBoth: in CallContextToString()
50 std::ostream& operator<<(std::ostream& out, const CallContext& context) { in operator <<()
55 CallContext GetInstructionCallContext(HloOpcode opcode) { in GetInstructionCallContext()
60 return CallContext::kSequential; in GetInstructionCallContext()
70 return CallContext::kParallel; in GetInstructionCallContext()
72 return CallContext::kNone; in GetInstructionCallContext()
[all …]
Dcall_graph.h32 enum class CallContext { enum
53 string CallContextToString(CallContext context);
54 std::ostream& operator<<(std::ostream& out, const CallContext& context);
56 CallContext GetInstructionCallContext(HloOpcode opcode);
63 CallContext context) in CallSite()
77 CallContext context() const { return context_; } in context()
89 const CallContext context_;
122 CallContext context() const { return context_; } in context()
136 void set_context(CallContext value) { context_ = value; } in set_context()
174 CallContext context_ = CallContext::kNone;
Dflatten_call_graph.cc77 if (call_site.context() == CallContext::kParallel) { in FlattenNode()
80 CHECK_EQ(call_site.context(), CallContext::kSequential); in FlattenNode()
84 if (node.context() != CallContext::kBoth && i == 0) { in FlattenNode()
100 CallContext::kSequential) { in FlattenNode()
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()
162 EXPECT_EQ(CallContext::kParallel, map_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()
239 EXPECT_EQ(CallContext::kParallel, map_callsite.context()); in TEST_F()
244 EXPECT_EQ(CallContext::kBoth, sub_node.context()); in TEST_F()
[all …]
Dconditional_to_select.cc72 if (node.context() != CallContext::kParallel) { in Run()
Dhlo_liveness_analysis.cc201 if (call_graph_node.context() == CallContext::kSequential) { in PropagateLivenessToParameterCallers()
231 if (call_graph_node.context() == CallContext::kSequential) { in PropagateLivenessThroughControlFlow()
Dhlo_dataflow_analysis.cc615 if (call_graph_node.context() == CallContext::kParallel || in UpdateParameterValueSet()
619 CHECK_EQ(call_graph_node.context(), CallContext::kSequential); in UpdateParameterValueSet()
891 if (call_graph_node.context() == CallContext::kSequential) { in Propagate()
977 if (call_graph_node.context() == CallContext::kBoth) { in InitializeInstructionValueSets()
989 call_graph_node.context() == CallContext::kParallel) { in InitializeInstructionValueSets()
Dcopy_insertion.cc1085 if (node.context() == CallContext::kParallel) { in AddSpecialCaseCopies()
1088 TF_RET_CHECK(node.context() == CallContext::kSequential); in AddSpecialCaseCopies()
Dhlo_rematerialization.cc1819 if (callsite == nullptr || callsite->context() == CallContext::kParallel) { in CalledComputationsMemoryUsage()
1934 callsite->context() == CallContext::kSequential && in RematerializeComputation()
2042 if (node.context() == CallContext::kSequential) { in Run()
Dmemory_space_assignment.cc1321 CallContext::kSequential); in AllocateAllocationValues()
/external/llvm/lib/Target/X86/
DX86CallFrameOptimization.cpp55 struct CallContext { struct in __anon49792e110111::X86CallFrameOptimization
56 CallContext() in CallContext() function
82 typedef SmallVector<CallContext, 8> ContextVector;
89 MachineBasicBlock::iterator I, CallContext &Context);
91 void adjustCallSequence(MachineFunction &MF, const CallContext &Context);
240 CallContext Context; in runOnMachineFunction()
320 CallContext &Context) { in collectCallInfo()
452 const CallContext &Context) { in adjustCallSequence()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/X86/
DX86CallFrameOptimization.cpp70 struct CallContext { struct in __anon3c1c13f30111::X86CallFrameOptimization
71 CallContext() : FrameSetup(nullptr), ArgStoreVector(4, nullptr) {} in CallContext() argument
95 typedef SmallVector<CallContext, 8> ContextVector;
102 MachineBasicBlock::iterator I, CallContext &Context);
104 void adjustCallSequence(MachineFunction &MF, const CallContext &Context);
260 CallContext Context; in runOnMachineFunction()
356 CallContext &Context) { in collectCallInfo()
494 const CallContext &Context) { in adjustCallSequence()
/external/llvm-project/llvm/lib/Target/X86/
DX86CallFrameOptimization.cpp71 struct CallContext { struct in __anonbca2735f0111::X86CallFrameOptimization
72 CallContext() : FrameSetup(nullptr), ArgStoreVector(4, nullptr) {} in CallContext() function
96 typedef SmallVector<CallContext, 8> ContextVector;
103 MachineBasicBlock::iterator I, CallContext &Context);
105 void adjustCallSequence(MachineFunction &MF, const CallContext &Context);
260 CallContext Context; in runOnMachineFunction()
356 CallContext &Context) { in collectCallInfo()
494 const CallContext &Context) { in adjustCallSequence()
/external/llvm-project/clang-tools-extra/clang-tidy/bugprone/
DNotNullTerminatedResultCheck.cpp677 struct CallContext { in registerMatchers() struct
678 CallContext(StringRef Name, Optional<unsigned> DestinationPos, in registerMatchers() function
691 auto MatchDestination = [=](CallContext CC) { in registerMatchers()
699 auto MatchSource = [=](CallContext CC) { in registerMatchers()
703 auto MatchGivenLength = [=](CallContext CC) { in registerMatchers()
721 auto MatchCall = [=](CallContext CC) { in registerMatchers()
734 auto Match = [=](CallContext CC) { in registerMatchers()
/external/llvm-project/llvm/lib/Transforms/IPO/
DSampleContextTracker.cpp368 ContextTrieNode *CallContext = getContextFor(DIL); in getCalleeContextFor() local
369 if (!CallContext) in getCalleeContextFor()
372 return CallContext->getChildContext( in getCalleeContextFor()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py429 call_ctx = CallContext()
439 class CallContext(object): class