Home
last modified time | relevance | path

Searched defs:computation (Results 1 – 25 of 383) sorted by relevance

12345678910>>...16

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module.cc76 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal()
131 std::unique_ptr<HloComputation> computation) { in AddEntryComputation()
138 std::unique_ptr<HloComputation> computation) { in AddEntryComputationWithLayouts()
160 std::unique_ptr<HloComputation> computation) { in AddEmbeddedComputation()
281 for (const HloComputation* computation : computations) { in ToCord() local
311 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto() local
366 for (const HloComputation* computation : computations()) { in CheckUniqueNamesAndIdsForComputationsAndInstructions() local
453 for (auto& computation : computations) { in CreateFromProto() local
622 const std::string& outlined_computation_name, HloComputation* computation) { in OutlineExpressionFromComputation()
708 for (const auto& computation : computations_) { in instruction_count() local
[all …]
H A Dhlo_schedule.cc43 for (const HloComputation* computation : module->computations()) { in CreateFromProto() local
57 const HloComputation* computation = comp_it->second; in CreateFromProto() local
94 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence()
99 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence()
108 const HloComputation* computation) { in GetOrCreateSequence()
127 const HloComputation* computation) { in UpdateComputationSchedule()
221 for (const HloComputation* computation : nonfusion_computations) { in Update() local
242 for (const HloComputation* computation : nonfusion_computations) { in Update() local
263 for (const HloComputation* computation : nonfusion_computations) { in Update() local
298 for (const HloComputation* computation : nonfusion_computations) { in Verify() local
[all …]
H A Dalgebraic_simplifier_test.cc70 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
375 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
534 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
556 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
576 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
601 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
662 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
688 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
733 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
769 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
H A Dhlo_creation_utils.cc48 HloComputation* computation = operand->parent(); in MakeUnaryHlo() local
63 HloComputation* computation = lhs->parent(); in MakeBinaryHlo() local
76 HloComputation* computation = lhs->parent(); in MakeCompareHlo() local
90 HloComputation* computation = operand->parent(); in MakePadHlo() local
107 HloComputation* computation = operand->parent(); in MakeSliceHlo() local
124 HloComputation* computation = lhs->parent(); in MakeConvolveHlo() local
164 HloComputation* computation = operand->parent(); in MakeDynamicSliceHlo() local
181 HloComputation* computation = operand->parent(); in MakeDynamicSliceHlo() local
210 HloComputation* computation = operand->parent(); in MakeDynamicUpdateSliceHlo() local
241 HloComputation* computation = operand->parent(); in MakeBroadcastHlo() local
[all …]
H A Dhlo_computation_test.cc111 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
129 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
162 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
184 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
209 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
213 explicit TestVisitor(HloComputation* computation) in TEST_F()
249 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
266 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
282 auto computation = builder.Build(); in TEST_F() local
[all …]
H A Dreshape_mover_test.cc55 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
98 auto computation = m->AddEntryComputation(builder.Build()); variable
124 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
151 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
205 auto computation = m->AddEntryComputation(builder.Build()); variable
242 auto computation = m->AddEntryComputation(builder.Build()); variable
286 auto computation = m->AddEntryComputation(builder.Build()); variable
332 auto computation = m->AddEntryComputation(builder.Build()); variable
375 auto computation = m->AddEntryComputation(builder.Build()); variable
403 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
H A Dhlo_cse_test.cc64 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
94 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
174 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
206 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
253 auto computation = m->entry_computation(); in TEST_F() local
302 auto computation = m->entry_computation(); in TEST_F() local
353 auto computation = m->entry_computation(); in TEST_F() local
394 auto computation = m->entry_computation(); in TEST_F() local
421 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
H A Dhlo_rematerialization_test.cc66 HloComputation* computation = in TEST_F() local
106 HloComputation* computation = in TEST_F() local
127 HloComputation* computation = in TEST_F() local
295 auto count_rngs = [](const HloComputation* computation) { in TEST_F()
383 auto count_broadcasts = [](const HloComputation* computation) { in TEST_F()
456 auto count_copies = [](const HloComputation* computation) { in TEST_F()
814 auto* computation = module->entry_computation(); in TEST_F() local
876 auto* computation = module->entry_computation(); in TEST_F() local
920 const HloComputation* computation = module->entry_computation(); in TEST_F() local
965 const HloComputation* computation = module->entry_computation(); in TEST_F() local
[all …]
H A Dbfloat16_normalization_test.cc113 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
141 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
171 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
209 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
268 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
293 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
322 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
355 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
381 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
420 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
H A Dbfloat16_propagation_test.cc126 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
186 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
225 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
248 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
304 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
343 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
372 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
415 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
448 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
481 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
H A Dcall_graph.cc85 [](std::string* out, const HloComputation* computation) { in ToString()
90 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode()
142 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode()
204 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
242 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
257 for (const HloComputation* computation : module_->computations()) { in SetNodeDepths() local
306 for (const HloComputation* computation : module->computations()) { in Build() local
333 for (const HloComputation* computation : node.callees()) { in VisitNodesInternal() local
H A Dconvolution_4d_expander_test.cc44 auto computation = module->entry_computation(); in TEST_F() local
69 auto computation = module->entry_computation(); in TEST_F() local
96 auto computation = module->entry_computation(); in TEST_F() local
121 auto computation = module->entry_computation(); in TEST_F() local
140 auto computation = module->entry_computation(); in TEST_F() local
159 auto computation = module->entry_computation(); in TEST_F() local
H A Dasync_collective_creator_test.cc53 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
78 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
105 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
137 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
167 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
196 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
H A Dhlo_dce_test.cc44 bool HasInstruction(const HloComputation& computation, in HasInstruction()
61 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
83 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
148 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
188 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
234 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
291 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
344 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
H A Dhlo_dce.cc65 HloComputation* computation, bool remove_cross_partition_collective_ops) { in RunOnComputation()
100 HloModule* module, HloComputation* computation, in RecursivelyRemoveDeadComputation()
141 for (auto* computation : in RecursivelyRemoveDeadComputations() local
152 for (auto* computation : in RecursivelyRemoveDeadComputations() local
177 for (auto* computation : in Run() local
H A Dhlo_memory_scheduler.cc81 HloComputation* computation, in Run()
106 ListScheduler(HloComputation* computation, in ListScheduler()
414 HloComputation* computation, in ScheduleComputationHelper()
437 HloComputation* computation, in DFSMemoryScheduler()
538 for (auto* computation : in ComputationSchedulerToModuleScheduler() local
559 HloComputation* computation, in ListMemoryScheduler()
582 HloComputation* computation, in PostOrderMemoryScheduler()
603 HloComputation* computation, in DefaultMemoryScheduler()
744 HloComputation* computation, const BufferValue::SizeFunction& size_function, in ScheduleComputation()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dbatch_ops_test.py221 def computation(in_t): function
270 def computation(in_t): function
295 def computation(in_t): function
344 def computation(in_t): function
374 def computation(in_t): function
408 def computation(inp): function
442 def computation(in0, in1): function
468 def computation(in_t): function
522 def computation(in_t): function
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_outside_compilation_test.py335 def computation(x): function
361 def computation(x): function
396 def computation(x): function
425 def computation(x): function
447 def computation(x, y): function
471 def computation(x, y): function
530 def computation(x): function
545 def computation(x): function
621 def computation(x): function
651 def computation(x): function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dhlo_control_flow_flattening.cc39 HloComputation* computation) { in CreateConstant()
73 const HloComputation* computation = while_hlo.parent(); in IsNotContainedInLoop() local
135 HloComputation* computation = while_hlo->parent(); in FlattenWhileLoop() local
254 HloComputation* computation = infeed_hlo->parent(); in RemoveInfeed() local
279 HloComputation* computation = recv_done->parent(); in RemoveRecvDone() local
301 HloComputation* computation = outfeed_hlo->parent(); in RemoveOutfeed() local
322 HloComputation* computation = send_done->parent(); in RemoveSendDone() local
338 HloComputation* computation = hlo->parent(); in RemoveCollective() local
354 HloComputation* computation = hlo->parent(); in RemovePartitionOrReplicaId() local
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcustom_training_loop_input_test.py87 def computation(x): function
256 def computation(x): function
499 def computation(x): function
521 def computation(x): function
549 def computation(inputs): function
574 def computation(x): function
598 def computation(x): function
651 def computation(x): function
759 def computation(x): function
880 def computation(x): function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_execute_test.cc129 auto computation = builder.Build().value(); in XLA_TEST_F() local
162 auto computation = builder.Build().value(); in XLA_TEST_F() local
201 auto computation = builder.Build().value(); in XLA_TEST_F() local
229 auto computation = builder.Build().value(); in XLA_TEST_F() local
302 auto computation = builder.Build().value(); in XLA_TEST_F() local
348 auto computation = builder.Build().value(); in XLA_TEST_F() local
380 auto computation = builder.Build().value(); in XLA_TEST_F() local
429 auto computation = builder.Build().value(); in XLA_TEST_F() local
481 auto computation = builder.Build().value(); in XLA_TEST_F() local
532 auto computation = builder.Build().value(); in XLA_TEST_F() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient.cc166 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in ExecuteAndTransfer()
183 StatusOr<Literal> Client::ComputeConstant(const XlaComputation& computation, in ComputeConstant()
217 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile()
277 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in Execute()
331 for (const XlaComputationInstance& computation : computations) { in ExecuteParallel() local
425 const XlaComputation& computation, in GetComputationStats()
446 const XlaComputation& computation) { in GetComputationShape()
468 const XlaComputation& computation, const ExecutionProfile& profile) { in ExecutionStatsAsString()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dinstruction_fusion_test.cc46 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
66 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
85 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
104 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
124 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
141 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
397 for (const auto* computation : module.computations()) { in Count() local
409 const HloComputation& computation, HloOpcode op) { in FindHloInstruction()
485 auto computation = module->AddEntryComputation(b.Build()); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc76 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
117 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
161 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
195 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
235 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
297 HloComputation* computation = module->AddEntryComputation(builder.Build()); in RunDotOutputFusion() local
340 const HloComputation* computation, in AssertCorrectLayoutForDotOutputFusion()
454 HloComputation* computation = module->entry_computation(); in TEST_F() local
/aosp_15_r20/external/tink/cc/mac/internal/
H A Dchunked_mac_test.cc78 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST_P() local
115 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local
144 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local
170 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local
198 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local

12345678910>>...16