| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule.cc | 36 for (const HloComputation* computation : module->computations()) { in CreateFromProto() local
37 id_to_computation[computation->unique_id()] = computation; in CreateFromProto()
47 const HloComputation* computation = comp_it->second; in CreateFromProto() local
50 for (HloInstruction* instruction : computation->instructions()) { in CreateFromProto()
55 schedule.GetOrCreateSequence(computation); in CreateFromProto()
59 << "No instruction exists in HLO computation " << computation->name() in CreateFromProto()
84 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence() argument
86 set_sequence(computation, HloInstructionSequence(sequence)); in set_sequence()
89 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence() argument
91 CHECK(computation->parent() == module_); in set_sequence()
[all …]
|
| D | hlo_module.cc | 68 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal() argument
72 entry_computation_ = computation.get(); in AddComputationInternal()
88 computation->UniquifyName(&computation_name_uniquer_); in AddComputationInternal()
89 for (auto* instruction : computation->instructions()) { in AddComputationInternal()
94 for (auto* instruction : computation->instructions()) { in AddComputationInternal()
98 CHECK_NE(computation->root_instruction()->unique_id(), -1) in AddComputationInternal()
99 << "Root has no valid id: " << computation->ToString(); in AddComputationInternal()
100 computation->SetUniqueId(computation->root_instruction()->unique_id()); in AddComputationInternal()
107 computation_name_uniquer_.GetUniqueName(computation->name()); in AddComputationInternal()
108 for (auto* instruction : computation->instructions()) { in AddComputationInternal()
[all …]
|
| D | root_instruction_sinker.cc | 24 void SinkTupleRoot(HloComputation* computation) { in SinkTupleRoot() argument
25 HloInstruction* root = computation->root_instruction(); in SinkTupleRoot()
30 computation->parent()->schedule().GetOrCreateSequence(computation); in SinkTupleRoot()
35 computation->set_root_instruction(new_root); in SinkTupleRoot()
39 void SinkNontupleRoot(HloComputation* computation) { in SinkNontupleRoot() argument
40 HloInstruction* root = computation->root_instruction(); in SinkNontupleRoot()
42 HloInstruction* new_root = computation->AddInstruction( in SinkNontupleRoot()
45 computation->parent()->schedule().GetOrCreateSequence(computation); in SinkNontupleRoot()
47 computation->set_root_instruction(new_root); in SinkNontupleRoot()
56 for (HloComputation* computation : module->MakeNonfusionComputations()) { in Run()
[all …]
|
| D | hlo_dce.cc | 41 HloComputation* computation, bool remove_cross_partition_collective_ops) { in RunOnComputation() argument
44 XLA_VLOG_LINES(3, computation->ToString()); in RunOnComputation()
49 for (auto* instruction : computation->instructions()) { in RunOnComputation()
51 if (instruction != computation->root_instruction() && in RunOnComputation()
53 computation->IsSafelyRemovable(instruction) && in RunOnComputation()
66 computation->RemoveInstructionAndUnusedOperands(dead_root)); in RunOnComputation()
71 XLA_VLOG_LINES(3, computation->ToString()); in RunOnComputation()
83 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
86 RunOnComputation(computation, remove_cross_partition_collective_ops_)); in Run()
96 for (auto* computation : module->MakeComputationPostOrder()) { in Run() local
[all …]
|
| D | computation_placer.cc | 77 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize() local
81 computation_device->add_replica_device_ids((*this)(replica, computation)); in Serialize()
98 for (int computation = 0; computation < proto.computation_count(); in Deserialize() local
99 ++computation) { in Deserialize()
100 const auto& computation_device = proto.computation_devices(computation); in Deserialize()
104 (*assignment)(replica, computation) = in Deserialize()
114 for (int computation = 0; computation < computation_count(); ++computation) { in ToString() local
115 StrAppend(&output, "Computation ", computation, ": "); in ToString()
117 StrAppend(&output, operator()(replica, computation), " "); in ToString()
124 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId() argument
[all …]
|
| D | comparison_expander.cc | 30 HloComputation* computation, HloInstruction* value, in BitcastConvertFloatingPointToIntegral() argument
46 auto signed_value = computation->AddInstruction( in BitcastConvertFloatingPointToIntegral()
48 auto unsigned_value = computation->AddInstruction( in BitcastConvertFloatingPointToIntegral()
50 auto flipped_value = computation->AddInstruction(HloInstruction::CreateBinary( in BitcastConvertFloatingPointToIntegral()
52 flipped_value = computation->AddInstruction( in BitcastConvertFloatingPointToIntegral()
56 auto is_negative = computation->AddInstruction(HloInstruction::CreateCompare( in BitcastConvertFloatingPointToIntegral()
58 return computation->AddInstruction( in BitcastConvertFloatingPointToIntegral()
82 HloComputation* computation = instruction->parent(); in ExpandInstruction() local
94 lhs = computation->AddInstruction( in ExpandInstruction()
96 rhs = computation->AddInstruction( in ExpandInstruction()
[all …]
|
| D | hlo_creation_utils.cc | 39 HloComputation* computation = operand->parent(); in MakeUnaryHlo() local
42 return computation->AddInstruction( in MakeUnaryHlo()
48 HloComputation* computation = lhs->parent(); in MakeBinaryHlo() local
49 CHECK_EQ(computation, rhs->parent()); in MakeBinaryHlo()
52 return computation->AddInstruction( in MakeBinaryHlo()
59 HloComputation* computation = lhs->parent(); in MakeCompareHlo() local
60 CHECK_EQ(computation, rhs->parent()); in MakeCompareHlo()
64 return computation->AddInstruction( in MakeCompareHlo()
71 HloComputation* computation = operand->parent(); in MakePadHlo() local
72 CHECK_EQ(computation, padding_value->parent()); in MakePadHlo()
[all …]
|
| D | hlo_cse_test.cc | 63 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
65 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
70 EXPECT_EQ(2, computation->instruction_count()); in TEST_F()
71 HloInstruction* constant = *computation->instructions().begin(); in TEST_F()
93 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
95 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
101 EXPECT_EQ(2, computation->instruction_count()); in TEST_F()
125 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
127 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
133 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
[all …]
|
| D | hlo_dce_test.cc | 46 bool HasInstruction(const HloComputation& computation, in HasInstruction() argument
48 return absl::c_linear_search(computation.instructions(), instruction); in HasInstruction()
63 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
65 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
70 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
85 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
87 EXPECT_EQ(5, computation->instruction_count()); in TEST_F()
92 EXPECT_EQ(5, computation->instruction_count()); in TEST_F()
150 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
152 EXPECT_EQ(5, computation->instruction_count()); in TEST_F()
[all …]
|
| D | hlo_memory_scheduler.cc | 81 HloComputation* computation, in Run() argument
86 ListScheduler scheduler(computation, points_to_analysis, size_function, in Run()
106 ListScheduler(HloComputation* computation, in ListScheduler() argument
111 : computation_(computation), in ListScheduler()
119 for (auto* instruction : computation->instructions()) { in ListScheduler()
135 for (auto* instruction : computation->instructions()) { in ListScheduler()
141 for (auto* instruction : computation->instructions()) { in ListScheduler()
150 points_to_analysis.GetPointsToSet(computation->root_instruction()) in ListScheduler()
414 HloComputation* computation, in ScheduleComputationHelper() argument
422 VLOG(2) << "Computation: " << computation->name(); in ScheduleComputationHelper()
[all …]
|
| D | reshape_mover_test.cc | 54 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
56 EXPECT_THAT(computation->root_instruction(), in TEST_F()
61 EXPECT_THAT(computation->root_instruction(), in TEST_F()
97 auto computation = m->AddEntryComputation(builder.Build()); variable
99 EXPECT_THAT(computation->root_instruction(),
104 EXPECT_THAT(computation->root_instruction(),
123 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
125 EXPECT_THAT(computation->root_instruction(), in TEST_F()
131 computation->root_instruction(), in TEST_F()
150 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
[all …]
|
| D | defuser_test.cc | 33 for (HloComputation* computation : m->computations()) { in FusionCount()
34 if (computation->IsFusionComputation()) { in FusionCount()
71 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
72 computation->CreateFusionInstruction({add}, in TEST_F()
75 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
81 EXPECT_THAT(computation->root_instruction(), in TEST_F()
97 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
98 computation->CreateFusionInstruction({add}, in TEST_F()
101 EXPECT_THAT(computation->root_instruction(), op::Negate(op::Fusion())); in TEST_F()
107 EXPECT_THAT(computation->root_instruction(), in TEST_F()
[all …]
|
| D | convolution_4d_expander_test.cc | 44 auto computation = module->entry_computation(); in TEST_F() local
45 HloInstruction* root = computation->root_instruction(); in TEST_F()
50 root = computation->root_instruction(); in TEST_F()
69 auto computation = module->entry_computation(); in TEST_F() local
70 HloInstruction* root = computation->root_instruction(); in TEST_F()
75 root = computation->root_instruction(); in TEST_F()
96 auto computation = module->entry_computation(); in TEST_F() local
97 HloInstruction* root = computation->root_instruction(); in TEST_F()
102 root = computation->root_instruction(); in TEST_F()
121 auto computation = module->entry_computation(); in TEST_F() local
[all …]
|
| D | hlo_computation_test.cc | 111 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
113 auto embedded_computations = computation->MakeEmbeddedComputationsList(); in TEST_F()
129 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
130 EXPECT_THAT(computation->MakeInstructionPostOrder(), ElementsAre(constant)); in TEST_F()
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
145 EXPECT_THAT(computation->MakeInstructionPostOrder(), in TEST_F()
161 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
163 EXPECT_THAT(computation->MakeInstructionPostOrder(), in TEST_F()
180 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
181 EXPECT_THAT(computation->MakeInstructionPostOrder(), in TEST_F()
[all …]
|
| D | hlo_schedule.h | 118 const HloComputation* computation) const;
123 const HloComputation* computation);
126 void set_sequence(const HloComputation* computation,
128 void set_sequence(const HloComputation* computation,
138 bool is_computation_scheduled(const HloComputation* computation) const { in is_computation_scheduled() argument
139 return sequences_.contains(computation->unique_id()); in is_computation_scheduled()
143 void remove_computation(const HloComputation* computation) { in remove_computation() argument
144 auto it = sequences_.find(computation->unique_id()); in remove_computation()
150 void remove_instruction(const HloComputation* computation, in remove_instruction() argument
152 sequences_[computation->unique_id()].remove_instruction(instruction); in remove_instruction()
[all …]
|
| D | flatten_call_graph.cc | 34 HloComputation* computation, in ReplaceCalledComputation() argument
38 if (computation == instruction->while_condition()) { in ReplaceCalledComputation()
41 CHECK_EQ(computation, instruction->while_body()); in ReplaceCalledComputation()
47 CHECK_EQ(instruction->to_apply(), computation); in ReplaceCalledComputation()
54 CHECK_EQ(computation, instruction->branch_computation(b)); in ReplaceCalledComputation()
56 if (computation == instruction->branch_computation(b)) { in ReplaceCalledComputation()
71 HloComputation* computation = node.computation(); in FlattenNode() local
72 HloModule* module = computation->parent(); in FlattenNode()
90 module->AddEmbeddedComputation(computation->Clone()); in FlattenNode()
91 ReplaceCalledComputation(call_site.instruction(), computation, clone); in FlattenNode()
|
| D | call_graph.cc | 81 [](string* out, const HloComputation* computation) { in ToString() argument
82 out->append(computation->name()); in ToString()
86 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode() argument
87 : computation_(computation) {} in CallGraphNode()
110 CHECK_EQ(instruction->parent(), computation()); in AddCallSiteForInstruction()
132 const HloComputation* computation) const { in GetNode()
133 auto it = node_indices_.find(computation); in GetNode()
138 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode() argument
139 auto it = node_indices_.find(computation); in GetNode()
200 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
[all …]
|
| D | convolution_group_converter_test.cc | 48 auto computation = module->entry_computation(); in TEST_F() local
49 HloInstruction* root = computation->root_instruction(); in TEST_F()
55 root = computation->root_instruction(); in TEST_F()
80 auto computation = module->entry_computation(); in TEST_F() local
81 HloInstruction* root = computation->root_instruction(); in TEST_F()
87 root = computation->root_instruction(); in TEST_F()
107 auto computation = module->entry_computation(); in TEST_F() local
108 HloInstruction* root = computation->root_instruction(); in TEST_F()
114 root = computation->root_instruction(); in TEST_F()
134 auto computation = module->entry_computation(); in TEST_F() local
[all …]
|
| D | loop_schedule_linearizer_test.cc | 39 int64 CountCopies(const HloComputation& computation) { in CountCopies() argument
41 for (const auto& instruction : computation.instructions()) { in CountCopies()
51 for (const auto& computation : module.computations()) { in CountCopies() local
52 count += CountCopies(*computation); in CountCopies()
57 int64 CountControlEdges(const HloComputation& computation) { in CountControlEdges() argument
59 for (const auto& instruction : computation.instructions()) { in CountControlEdges()
67 for (const auto& computation : module.computations()) { in CountControlEdges() local
68 count += CountControlEdges(*computation); in CountControlEdges()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | compilation_cache_test.cc | 41 void ExecuteComputationR0F32(const XlaComputation& computation, in ExecuteComputationR0F32() argument
47 ->ExecuteAndTransfer(computation, arguments, in ExecuteComputationR0F32()
57 const XlaComputation& computation, in ExecuteComputationR2F32() argument
63 ->Execute(computation, arguments, in ExecuteComputationR2F32()
79 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F() local
81 ExecuteComputationR0F32(computation, {}, -42.0, /*expect_cache_hit=*/false); in XLA_TEST_F()
82 ExecuteComputationR0F32(computation, {}, -42.0, /*expect_cache_hit=*/true); in XLA_TEST_F()
83 ExecuteComputationR0F32(computation, {}, -42.0, /*expect_cache_hit=*/true); in XLA_TEST_F()
101 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F() local
103 ExecuteComputationR0F32(computation, {data_42.get()}, -42.0, in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | test.cc | 54 void zero_buffers(XlaCompiledCpuFunction* computation) { in zero_buffers() argument
55 for (int i = 0; i < computation->num_args(); ++i) { in zero_buffers()
56 memset(computation->arg_data(i), 0, computation->arg_size(i)); in zero_buffers()
65 CPP_CLASS computation; in TEST() local
66 computation.set_thread_pool(&device); in TEST()
67 zero_buffers(&computation); in TEST()
69 EXPECT_TRUE(computation.Run()); in TEST()
79 CPP_CLASS computation; in BM_NAME() local
80 computation.set_thread_pool(&device); in BM_NAME()
81 zero_buffers(&computation); in BM_NAME()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_executable.cc | 118 for (int computation = 0; computation < args.size(); ++computation) { in ExecuteOnLocalDevices() local
119 arg_buffers[computation].resize(args[computation].size()); in ExecuteOnLocalDevices()
120 absl::c_transform(args[computation], arg_buffers[computation].begin(), in ExecuteOnLocalDevices()
129 for (int computation = 0; computation < output_buffers.size(); in ExecuteOnLocalDevices() local
130 ++computation) { in ExecuteOnLocalDevices()
131 for (auto& buffer : output_buffers[computation]) { in ExecuteOnLocalDevices()
132 outputs[computation].push_back( in ExecuteOnLocalDevices()
160 for (int computation = 0; computation < num_computations; ++computation) { in ExecuteShardedOnLocalDevices() local
161 arg_buffers[computation].resize(args.size()); in ExecuteShardedOnLocalDevices()
162 absl::c_transform(args, arg_buffers[computation].begin(), in ExecuteShardedOnLocalDevices()
[all …]
|
| /external/llvm-project/clang/lib/AST/ |
| D | Linkage.h | 107 LVComputationKind computation);
110 LVComputationKind computation);
114 LVComputationKind computation);
118 LVComputationKind computation);
122 LVComputationKind computation);
125 LVComputationKind computation,
129 LVComputationKind computation,
133 LVComputationKind computation);
136 LVComputationKind computation);
138 LinkageInfo getLVForType(const Type &T, LVComputationKind computation);
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 53 HloComputation* computation = operand->parent(); in CreateCholesky() local
92 computation->AddInstruction(HloInstruction::CreateCustomCall( in CreateCholesky()
96 HloInstruction* out = computation->AddInstruction( in CreateCholesky()
98 HloInstruction* info = computation->AddInstruction( in CreateCholesky()
103 HloInstruction* zero = computation->AddInstruction( in CreateCholesky()
106 computation->AddInstruction(HloInstruction::CreateBroadcast( in CreateCholesky()
108 HloInstruction* ok = computation->AddInstruction( in CreateCholesky()
111 ok = computation->AddInstruction(HloInstruction::CreateBroadcast( in CreateCholesky()
117 HloInstruction* nan = computation->AddInstruction( in CreateCholesky()
120 computation->AddInstruction(HloInstruction::CreateBroadcast( in CreateCholesky()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.h | 62 const XlaComputation& computation,
90 const XlaComputation& computation,
101 const XlaComputation& computation; member
106 XlaComputationInstance(const XlaComputation& computation, in XlaComputationInstance()
110 : computation(computation), in XlaComputationInstance()
171 const XlaComputation& computation,
192 const XlaComputation& computation,
204 const XlaComputation& computation,
214 const XlaComputation& computation);
232 StatusOr<string> ExecutionStatsAsString(const XlaComputation& computation,
|