Home
last modified time | relevance | path

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

12345678910>>...12

/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.cc70 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal()
125 std::unique_ptr<HloComputation> computation) { in AddEntryComputation()
132 std::unique_ptr<HloComputation> computation) { in AddEntryComputationWithLayouts()
154 std::unique_ptr<HloComputation> computation) { in AddEmbeddedComputation()
259 for (const HloComputation* computation : computations) { in ToString() local
283 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto() local
314 for (const HloComputation* computation : computations()) { in CheckUniqueNamesAndIdsForComputationsAndInstructions() local
401 for (auto& computation : computations) { in CreateFromProto() local
515 const string& outlined_computation_name, HloComputation* computation) { in OutlineExpressionFromComputation()
601 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()
82 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
95 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
104 for (auto* computation : module->MakeComputationPostOrder()) { in Run() 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 …]
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
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 …]
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
Dcomputation_placer.cc78 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize() local
99 for (int computation = 0; computation < proto.computation_count(); in Deserialize() local
115 for (int computation = 0; computation < computation_count(); ++computation) { in ToString() local
125 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
138 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.cc83 [](string* out, const HloComputation* computation) { in ToString()
88 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode()
140 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode()
202 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
240 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
255 for (const HloComputation* computation : module_->computations()) { in SetNodeDepths() local
304 for (const HloComputation* computation : module->computations()) { in Build() local
331 for (const HloComputation* computation : node.callees()) { in VisitNodesInternal() local
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
634 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
660 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
705 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
741 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
788 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
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()
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
Dhlo_ordering.h131 const HloComputation& computation) const override { in SequentialOrder()
135 HloReachabilityMap& reachability_map(const HloComputation* computation) { in reachability_map()
139 const HloComputation* computation) const { in reachability_map()
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()
773 auto* computation = module->entry_computation(); in TEST_F() local
835 auto* computation = module->entry_computation(); in TEST_F() local
879 const HloComputation* computation = module->entry_computation(); in TEST_F() local
924 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/xla/tools/
Dhlo_control_flow_flattening.cc29 HloComputation* computation) { in CreateConstant()
70 HloComputation* computation = while_hlo->parent(); in FlattenWhileLoop() local
155 HloComputation* computation = infeed_hlo->parent(); in RemoveInfeed() local
174 HloComputation* computation = outfeed_hlo->parent(); in RemoveOutfeed() local
188 HloComputation* computation = hlo->parent(); in RemoveAllReduce() local
/external/tensorflow/tensorflow/python/tpu/
Dtpu_outside_compilation_test.py336 def computation(x): function
362 def computation(x): function
397 def computation(x): function
426 def computation(x): function
448 def computation(x, y): function
472 def computation(x, y): function
531 def computation(x): function
546 def computation(x): function
565 def computation(x): function
595 def computation(x): function
[all …]
/external/tensorflow/tensorflow/compiler/aot/
Dtest.cc54 void zero_buffers(XlaCompiledCpuFunction* computation) { in zero_buffers()
65 CPP_CLASS computation; in TEST() local
77 CPP_CLASS computation; in BM_NAME() local
/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()

12345678910>>...12