Home
last modified time | relevance | path

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

12345678910>>...13

/external/tensorflow/tensorflow/compiler/xla/service/
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 …]
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 …]
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.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 …]
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
Dloop_schedule_linearizer_test.cc39 int64_t CountCopies(const HloComputation& computation) { in CountCopies()
51 for (const auto& computation : module.computations()) { in CountCopies() local
57 int64_t CountControlEdges(const HloComputation& computation) { in CountControlEdges()
67 for (const auto& computation : module.computations()) { in CountControlEdges() local
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 …]
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 …]
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
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 …]
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.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
Dcomputation_placer.cc91 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize() local
112 for (int computation = 0; computation < proto.computation_count(); in Deserialize() local
128 for (int computation = 0; computation < computation_count(); ++computation) { in ToString() local
138 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
151 for (int computation = 0; computation < computation_count; ++computation) { in AssignDevices() local
Dtuple_util.cc26 HloComputation* computation = input_tuple->parent(); in ExtractPrefix() local
46 HloComputation* computation = input_tuple->parent(); in AppendSuffix() local
90 HloComputation* computation = new_instruction->parent(); in ReplaceTupleWith() local
149 HloComputation* computation = instruction->parent(); in AddGetTupleElements() local
Dhlo_schedule.h145 bool is_computation_scheduled(const HloComputation* computation) const { in is_computation_scheduled()
150 void remove_computation(const HloComputation* computation) { in remove_computation()
158 void remove_instruction(const HloComputation* computation, in remove_instruction()
165 void replace_instruction(const HloComputation* computation, in replace_instruction()
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_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 …]
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 …]
/external/tensorflow/tensorflow/python/ops/
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
/external/tensorflow/tensorflow/compiler/xla/tools/
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
/external/tink/cc/mac/internal/
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
/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().value(); in XLA_TEST_F() local
98 XlaComputation computation = builder.Build().value(); in XLA_TEST_F() local
151 XlaComputation computation = builder.Build().value(); in XLA_TEST_F() local
/external/tensorflow/tensorflow/python/tpu/
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 …]
/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.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()

12345678910>>...13