Home
last modified time | relevance | path

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

123456789

/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_creation_utils.cc37 HloComputation* computation = lhs->parent(); in MakeBinaryHlo() local
48 HloComputation* computation = lhs->parent(); in MakeCompareHlo() local
60 HloComputation* computation = operand->parent(); in MakePadHlo() local
74 HloComputation* computation = operand->parent(); in MakeSliceHlo() local
86 HloComputation* computation = lhs->parent(); in MakeConvolveHlo() local
99 HloComputation* computation = operand->parent(); in MakeTransposeHlo() local
109 HloComputation* computation = operand->parent(); in MakeReshapeHlo() local
124 HloComputation* computation = operand->parent(); in MakeDynamicSliceHlo() local
151 HloComputation* computation = operand->parent(); in MakeDynamicUpdateSliceHlo() local
179 HloComputation* computation = operand->parent(); in MakeBroadcastHlo() local
[all …]
Dhlo_module.cc52 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal()
104 std::unique_ptr<HloComputation> computation) { in AddEntryComputation()
120 std::unique_ptr<HloComputation> computation) { in AddEmbeddedComputation()
208 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToString() local
229 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto() local
250 for (const HloComputation* computation : computations()) { in CheckUniqueNamesAndIdsForComputationsAndInstructions() local
335 for (auto& computation : computations) { in CreateFromProto() local
407 const string& outlined_computation_name, HloComputation* computation) { in OutlineExpressionFromComputation()
493 for (const auto& computation : computations_) { in instruction_count() local
504 for (auto& computation : computations_) { in MakeComputationPostOrder() local
[all …]
Dhlo_dce.cc44 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
73 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
82 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
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()
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 …]
Dcomputation_placer.cc44 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize() local
65 for (int computation = 0; computation < proto.computation_count(); in Deserialize() local
81 for (int computation = 0; computation < computation_count(); ++computation) { in ToString() local
91 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
104 for (int computation = 0; computation < computation_count; ++computation) { in AssignDevices() local
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 …]
Dhlo_computation_test.cc112 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
130 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
145 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
162 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
181 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
203 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
228 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
232 explicit TestVisitor(HloComputation* computation) in TEST_F()
268 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
285 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dconditional_simplifier_test.cc100 HloComputation* computation = MakeConditional(m.get()); in TEST_F() local
108 HloComputation* computation = MakeConditional(m.get()); in TEST_F() local
120 HloComputation* computation = MakeConditional(m.get()); in TEST_F() local
136 HloComputation* computation = MakeConditional(m.get()); in TEST_F() local
150 HloComputation* computation = MakeConditional(m.get()); in TEST_F() local
Dcall_graph.cc80 [](string* out, const HloComputation* computation) { in ToString()
85 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode()
135 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode()
197 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
235 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
250 for (const HloComputation* computation : module_->computations()) { in SetNodeDepths() local
299 for (const HloComputation* computation : module->computations()) { in Build() local
326 for (const HloComputation* computation : node.callees()) { in VisitNodesInternal() local
Dalgebraic_simplifier_test.cc67 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
290 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
312 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
334 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
354 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
412 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
438 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
462 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
498 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
545 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dconvolution_group_converter_test.cc49 auto computation = module->entry_computation(); in TEST_F() local
80 auto computation = module->entry_computation(); in TEST_F() local
109 auto computation = module->entry_computation(); in TEST_F() local
Dhlo_dce_test.cc44 bool HasInstruction(const HloComputation& computation, in HasInstruction()
61 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
82 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
112 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
152 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
198 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
253 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
308 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
Dreduce_precision_insertion_test.cc58 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
85 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
115 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
177 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
209 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
273 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
299 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
325 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
352 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
Dbfloat16_normalization_test.cc111 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
139 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
169 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
207 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
264 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
294 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
321 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
360 auto computation = module->AddEntryComputation(builder.Build()); 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_execution_profile.h53 size_t GetProfileIndexFor(const HloComputation& computation) const { in GetProfileIndexFor()
121 uint64 total_cycles_executed(const HloComputation& computation) const { in total_cycles_executed()
127 void set_total_cycles_executed(const HloComputation& computation, in set_total_cycles_executed()
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/contrib/batching/python/ops/
Dbatch_ops_test.py196 def computation(in_t): function
221 def computation(in_t): function
246 def computation(in_t): function
268 def computation(in_t): function
300 def computation(inp): function
332 def computation(in0, in1): function
354 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.cc49 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
74 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
128 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc45 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
65 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
84 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
103 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
121 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
138 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
471 for (const auto* computation : module.computations()) { in Count() local
483 const HloComputation& computation, HloOpcode op) { in FindHloInstruction()
559 auto computation = module->AddEntryComputation(b.Build()); in TEST_F() local

123456789