Home
last modified time | relevance | path

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

1234567891011

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_schedule.cc36 for (const HloComputation* computation : module->computations()) { in CreateFromProto() local
47 const HloComputation* computation = comp_it->second; in CreateFromProto() local
84 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence()
89 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence()
96 const HloComputation* computation) { in GetOrCreateSequence()
113 const HloComputation* computation) { in UpdateComputationSchedule()
206 for (const HloComputation* computation : nonfusion_computations) { in Update() local
214 for (const HloComputation* computation : nonfusion_computations) { in Update() local
227 for (const HloComputation* computation : nonfusion_computations) { in Update() local
246 for (const HloComputation* computation : nonfusion_computations) { in Verify() local
[all …]
Dhlo_module.cc68 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal()
123 std::unique_ptr<HloComputation> computation) { in AddEntryComputation()
130 std::unique_ptr<HloComputation> computation) { in AddEntryComputationWithLayouts()
152 std::unique_ptr<HloComputation> computation) { in AddEmbeddedComputation()
253 for (const HloComputation* computation : computations) { in ToString() local
277 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto() local
307 for (const HloComputation* computation : computations()) { in CheckUniqueNamesAndIdsForComputationsAndInstructions() local
394 for (auto& computation : computations) { in CreateFromProto() local
508 const string& outlined_computation_name, HloComputation* computation) { in OutlineExpressionFromComputation()
594 for (const auto& computation : computations_) { in instruction_count() local
[all …]
Dhlo_creation_utils.cc39 HloComputation* computation = operand->parent(); in MakeUnaryHlo() local
48 HloComputation* computation = lhs->parent(); in MakeBinaryHlo() local
59 HloComputation* computation = lhs->parent(); in MakeCompareHlo() local
71 HloComputation* computation = operand->parent(); in MakePadHlo() local
85 HloComputation* computation = operand->parent(); in MakeSliceHlo() local
99 HloComputation* computation = lhs->parent(); in MakeConvolveHlo() local
113 HloComputation* computation = operand->parent(); in MakeTransposeHlo() local
123 HloComputation* computation = operand->parent(); in MakeReshapeHlo() local
138 HloComputation* computation = operand->parent(); in MakeDynamicSliceHlo() local
153 HloComputation* computation = operand->parent(); in MakeDynamicSliceHlo() local
[all …]
Dhlo_dce.cc41 HloComputation* computation, bool remove_cross_partition_collective_ops) { in RunOnComputation()
83 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
96 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
105 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
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
Dreshape_mover_test.cc54 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
97 auto computation = m->AddEntryComputation(builder.Build()); variable
123 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
150 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
204 auto computation = m->AddEntryComputation(builder.Build()); variable
241 auto computation = m->AddEntryComputation(builder.Build()); variable
285 auto computation = m->AddEntryComputation(builder.Build()); variable
331 auto computation = m->AddEntryComputation(builder.Build()); variable
374 auto computation = m->AddEntryComputation(builder.Build()); variable
402 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dhlo_cse_test.cc63 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
93 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
125 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
175 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
205 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
237 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
283 auto computation = m->entry_computation(); in TEST_F() local
333 auto computation = m->entry_computation(); in TEST_F() local
375 auto computation = m->entry_computation(); in TEST_F() local
416 auto computation = m->entry_computation(); in TEST_F() local
[all …]
Dloop_schedule_linearizer_test.cc39 int64 CountCopies(const HloComputation& computation) { in CountCopies()
51 for (const auto& computation : module.computations()) { in CountCopies() local
57 int64 CountControlEdges(const HloComputation& computation) { in CountControlEdges()
67 for (const auto& computation : module.computations()) { in CountControlEdges() local
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
161 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
180 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
202 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
227 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
231 explicit TestVisitor(HloComputation* computation) in TEST_F()
267 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
284 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dcomputation_placer.cc77 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize() local
98 for (int computation = 0; computation < proto.computation_count(); in Deserialize() local
114 for (int computation = 0; computation < computation_count(); ++computation) { in ToString() local
124 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
137 for (int computation = 0; computation < computation_count; ++computation) { in AssignDevices() local
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 …]
Dcall_graph.cc81 [](string* out, const HloComputation* computation) { in ToString()
86 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode()
138 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode()
200 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
238 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
253 for (const HloComputation* computation : module_->computations()) { in SetNodeDepths() local
302 for (const HloComputation* computation : module->computations()) { in Build() local
329 for (const HloComputation* computation : node.callees()) { in VisitNodesInternal() local
Dhlo_schedule.h138 bool is_computation_scheduled(const HloComputation* computation) const { in is_computation_scheduled()
143 void remove_computation(const HloComputation* computation) { in remove_computation()
150 void remove_instruction(const HloComputation* computation, in remove_instruction()
157 void replace_instruction(const HloComputation* computation, in replace_instruction()
Dhlo_ordering.h122 const HloComputation& computation) const override { in SequentialOrder()
126 HloReachabilityMap& reachability_map(const HloComputation* computation) { in reachability_map()
130 const HloComputation* computation) const { in reachability_map()
Dconvolution_group_converter_test.cc48 auto computation = module->entry_computation(); in TEST_F() local
80 auto computation = module->entry_computation(); in TEST_F() local
107 auto computation = module->entry_computation(); in TEST_F() local
134 auto computation = module->entry_computation(); in TEST_F() local
Dalgebraic_simplifier_test.cc69 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
374 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
533 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
555 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
575 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
633 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
659 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
704 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
740 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
787 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dhlo_dce_test.cc46 bool HasInstruction(const HloComputation& computation, in HasInstruction()
63 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
85 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
150 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
190 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
236 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
293 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
346 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
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()
772 auto* computation = module->entry_computation(); in TEST_F() local
834 auto* computation = module->entry_computation(); in TEST_F() local
878 const HloComputation* computation = module->entry_computation(); in TEST_F() local
923 const HloComputation* computation = module->entry_computation(); in TEST_F() local
[all …]
Ddefuser_test.cc71 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
97 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
135 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
174 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
202 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
/external/tensorflow/tensorflow/python/ops/
Dbatch_ops_test.py222 def computation(in_t): function
251 def computation(in_t): function
276 def computation(in_t): function
325 def computation(in_t): function
355 def computation(in_t): function
389 def computation(inp): function
423 def computation(in0, in1): function
449 def computation(in_t): function
503 def computation(in_t): function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcompilation_cache_test.cc41 void ExecuteComputationR0F32(const XlaComputation& computation, in ExecuteComputationR0F32()
57 const XlaComputation& computation, in ExecuteComputationR2F32()
79 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F() local
101 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F() local
156 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/aot/
Dtest.cc54 void zero_buffers(XlaCompiledCpuFunction* computation) { in zero_buffers()
65 CPP_CLASS computation; in TEST() local
79 CPP_CLASS computation; in BM_NAME() local
/external/tensorflow/tensorflow/python/tpu/
Dtpu_outside_compilation_test.py337 def computation(x): function
363 def computation(x): function
398 def computation(x): function
427 def computation(x): function
449 def computation(x, y): function
473 def computation(x, y): function
532 def computation(x): function
547 def computation(x): function
566 def computation(x): function
595 def computation(x): function
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc167 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in ExecuteAndTransfer()
184 StatusOr<Literal> Client::ComputeConstant(const XlaComputation& computation, in ComputeConstant()
218 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile()
278 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in Execute()
332 for (const XlaComputationInstance& computation : computations) { in ExecuteParallel() local
424 const XlaComputation& computation, in GetComputationStats()
445 const XlaComputation& computation) { in GetComputationShape()
467 const XlaComputation& computation, const ExecutionProfile& profile) { in ExecutionStatsAsString()
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_compilation_test.py64 def computation(): function
72 def computation(): function

1234567891011