Home
last modified time | relevance | path

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

12345678910

/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_dce.cc38 StatusOr<bool> HloDCE::RunOnComputation(HloComputation* computation) { in RunOnComputation()
76 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
88 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
97 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
Dhlo_module.cc63 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal()
115 std::unique_ptr<HloComputation> computation) { in AddEntryComputation()
136 std::unique_ptr<HloComputation> computation) { in AddEmbeddedComputation()
229 for (const HloComputation* computation : computations) { in ToString() local
253 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto() local
274 for (const HloComputation* computation : computations()) { in CheckUniqueNamesAndIdsForComputationsAndInstructions() local
361 for (auto& computation : computations) { in CreateFromProto() local
463 const string& outlined_computation_name, HloComputation* computation) { in OutlineExpressionFromComputation()
549 for (const auto& computation : computations_) { in instruction_count() local
560 for (auto& computation : computations_) { in MakeComputationPostOrder() local
[all …]
Dhlo_creation_utils.cc38 HloComputation* computation = lhs->parent(); in MakeBinaryHlo() local
49 HloComputation* computation = lhs->parent(); in MakeCompareHlo() local
61 HloComputation* computation = operand->parent(); in MakePadHlo() local
75 HloComputation* computation = operand->parent(); in MakeSliceHlo() local
87 HloComputation* computation = lhs->parent(); in MakeConvolveHlo() local
100 HloComputation* computation = operand->parent(); in MakeTransposeHlo() local
110 HloComputation* computation = operand->parent(); in MakeReshapeHlo() local
125 HloComputation* computation = operand->parent(); in MakeDynamicSliceHlo() local
140 HloComputation* computation = operand->parent(); in MakeDynamicSliceHlo() local
167 HloComputation* computation = operand->parent(); in MakeDynamicUpdateSliceHlo() 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
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_ordering.h101 const HloComputation& computation) const override { in SequentialOrder()
105 HloReachabilityMap& reachability_map(const HloComputation* computation) { in reachability_map()
109 const HloComputation* computation) const { in reachability_map()
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 …]
Dcomputation_placer.cc68 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize() local
89 for (int computation = 0; computation < proto.computation_count(); in Deserialize() local
105 for (int computation = 0; computation < computation_count(); ++computation) { in ToString() local
115 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
128 for (int computation = 0; computation < computation_count; ++computation) { in AssignDevices() 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 …]
Dhlo_get_dimension_size_rewriter.cc35 HloComputation* computation = instr->parent(); in ReplaceGetSize() local
97 for (auto* computation : module->computations()) { in Run() local
104 for (auto* computation : module->computations()) { in Run() 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()
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
267 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
292 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
321 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
354 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
380 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
419 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dcall_graph.cc80 [](string* out, const HloComputation* computation) { in ToString()
85 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode()
137 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode()
199 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
237 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
252 for (const HloComputation* computation : module_->computations()) { in SetNodeDepths() local
301 for (const HloComputation* computation : module->computations()) { in Build() local
328 for (const HloComputation* computation : node.callees()) { in VisitNodesInternal() local
Dalgebraic_simplifier_test.cc69 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
292 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
378 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
400 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
420 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
478 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
504 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
549 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
585 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
632 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dhlo_creation_utils.h164 StatusOr<HloInstruction*> MakeR1ConstantHlo(HloComputation* computation, in MakeR1ConstantHlo()
178 HloInstruction* MakeR0ConstantHlo(HloComputation* computation, NativeT value) { in MakeR0ConstantHlo()
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
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
Dtuple_util.cc26 HloComputation* computation = input_tuple->parent(); in ExtractPrefix() local
46 HloComputation* computation = input_tuple->parent(); in AppendSuffix() local
Dhlo_rematerialization_test_utils_test.cc39 auto computation = MakeRematerializableComputation(); in TEST_F() local
72 auto computation = MakeRematerializableWhileComputation( in TEST_F() local
/external/tensorflow/tensorflow/python/ops/
Dbatch_ops_test.py215 def computation(in_t): function
242 def computation(in_t): function
266 def computation(in_t): function
300 def computation(inp): function
334 def computation(in0, in1): function
358 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/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/compiler/xla/service/gpu/tests/
Dgpu_ldg_test.cc54 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
84 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
143 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local

12345678910