| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule.cc | 43 for (const HloComputation* computation : module->computations()) { in CreateFromProto() local
44 id_to_computation[computation->unique_id()] = computation; in CreateFromProto()
57 const HloComputation* computation = comp_it->second; in CreateFromProto() local
60 for (HloInstruction* instruction : computation->instructions()) { in CreateFromProto()
65 schedule.GetOrCreateSequence(computation); in CreateFromProto()
69 << "No instruction exists in HLO computation " << computation->name() in CreateFromProto()
94 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence() argument
96 set_sequence(computation, HloInstructionSequence(sequence)); in set_sequence()
99 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence() argument
101 CHECK(computation->parent() == module_); in set_sequence()
[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()
58 for (HloComputation* computation : in Run()
[all …]
|
| D | hlo_module.cc | 76 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal() argument
80 entry_computation_ = computation.get(); in AddComputationInternal()
96 computation->UniquifyName(&computation_name_uniquer_); in AddComputationInternal()
97 for (auto* instruction : computation->instructions()) { in AddComputationInternal()
102 for (auto* instruction : computation->instructions()) { in AddComputationInternal()
106 CHECK_NE(computation->root_instruction()->unique_id(), -1) in AddComputationInternal()
107 << "Root has no valid id: " << computation->ToString(); in AddComputationInternal()
108 computation->SetUniqueId(computation->root_instruction()->unique_id()); in AddComputationInternal()
115 computation_name_uniquer_.GetUniqueName(computation->name()); in AddComputationInternal()
116 for (auto* instruction : computation->instructions()) { in AddComputationInternal()
[all …]
|
| D | hlo_dce.cc | 49 for (HloComputation* computation : instruction->called_computations()) { in IsRemovableWhile()
50 for (HloInstruction* called_instr : computation->instructions()) { in IsRemovableWhile()
65 HloComputation* computation, bool remove_cross_partition_collective_ops) { in RunOnComputation() argument
68 XLA_VLOG_LINES(3, computation->ToString()); in RunOnComputation()
73 for (auto* instruction : computation->instructions()) { in RunOnComputation()
75 if (instruction->IsDead() && computation->IsSafelyRemovable(instruction) && in RunOnComputation()
89 computation->RemoveInstructionAndUnusedOperands(dead_root)); in RunOnComputation()
94 XLA_VLOG_LINES(3, computation->ToString()); in RunOnComputation()
100 HloModule* module, HloComputation* computation, in RecursivelyRemoveDeadComputation() argument
103 for (HloInstruction* instruction : computation->instructions()) { in RecursivelyRemoveDeadComputation()
[all …]
|
| D | async_collective_creator_test.cc | 53 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
54 ASSERT_THAT(computation, NotNull()); in TEST_F()
55 ASSERT_EQ(computation->instruction_count(), 3); in TEST_F()
56 const HloInstruction* done = computation->root_instruction(); in TEST_F()
78 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
79 ASSERT_THAT(computation, NotNull()); in TEST_F()
80 ASSERT_EQ(computation->instruction_count(), 3); in TEST_F()
81 const HloInstruction* done = computation->root_instruction(); in TEST_F()
105 HloComputation* computation = hlo_module->entry_computation(); in TEST_F() local
106 ASSERT_THAT(computation, NotNull()); in TEST_F()
[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()
66 computation->RemoveAsyncInstruction(instruction); in ReplaceCalledComputation()
80 HloComputation* computation = node.computation(); in FlattenNode() local
86 HloModule* module = computation->parent(); in FlattenNode()
106 if (computation->IsAsyncComputation()) { in FlattenNode()
[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 | computation_placer.cc | 91 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize() local
95 computation_device->add_replica_device_ids((*this)(replica, computation)); in Serialize()
112 for (int computation = 0; computation < proto.computation_count(); in Deserialize() local
113 ++computation) { in Deserialize()
114 const auto& computation_device = proto.computation_devices(computation); in Deserialize()
118 (*assignment)(replica, computation) = in Deserialize()
128 for (int computation = 0; computation < computation_count(); ++computation) { in ToString() local
129 StrAppend(&output, "Computation ", computation, ": "); in ToString()
131 StrAppend(&output, operator()(replica, computation), " "); in ToString()
138 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId() argument
[all …]
|
| D | collective_decomposer_utils.cc | 37 int64_t shard_dimension, HloComputation *computation, in CreateStartIndicesForCollectiveDecomposition() argument
39 HloInstruction *zero = computation->AddInstruction( in CreateStartIndicesForCollectiveDecomposition()
50 return computation->AddInstruction(HloInstruction::CreatePartitionId()); in CreateStartIndicesForCollectiveDecomposition()
52 const HloModuleConfig &config = computation->parent()->config(); in CreateStartIndicesForCollectiveDecomposition()
54 computation->AddInstruction(HloInstruction::CreateConstant( in CreateStartIndicesForCollectiveDecomposition()
56 HloInstruction *mul = computation->AddInstruction( in CreateStartIndicesForCollectiveDecomposition()
59 return computation->AddInstruction(HloInstruction::CreateBinary( in CreateStartIndicesForCollectiveDecomposition()
61 computation->AddInstruction(HloInstruction::CreatePartitionId()))); in CreateStartIndicesForCollectiveDecomposition()
68 computation->AddInstruction(HloInstruction::CreateReplicaId()); in CreateStartIndicesForCollectiveDecomposition()
82 computation->AddInstruction(HloInstruction::CreateReplicaId()); in CreateStartIndicesForCollectiveDecomposition()
[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 | 55 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
57 EXPECT_THAT(computation->root_instruction(), in TEST_F()
62 EXPECT_THAT(computation->root_instruction(), in TEST_F()
98 auto computation = m->AddEntryComputation(builder.Build()); variable
100 EXPECT_THAT(computation->root_instruction(),
105 EXPECT_THAT(computation->root_instruction(),
124 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
126 EXPECT_THAT(computation->root_instruction(), in TEST_F()
132 computation->root_instruction(), in TEST_F()
151 auto computation = m->AddEntryComputation(builder.Build()); 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()
162 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
163 EXPECT_THAT(computation->MakeInstructionPostOrder(), in TEST_F()
184 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
185 auto post_order = computation->MakeInstructionPostOrder(); in TEST_F()
[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 | hlo_cse_test.cc | 64 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
66 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
71 EXPECT_EQ(2, computation->instruction_count()); in TEST_F()
72 HloInstruction* constant = *computation->instructions().begin(); in TEST_F()
94 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
96 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
102 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
146 EXPECT_EQ(20, computation->instruction_count()); in TEST_F()
153 EXPECT_EQ(18, computation->instruction_count()); 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_dce_test.cc | 44 bool HasInstruction(const HloComputation& computation, in HasInstruction() argument
46 return absl::c_linear_search(computation.instructions(), instruction); in HasInstruction()
61 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
63 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
68 EXPECT_EQ(3, computation->instruction_count()); in TEST_F()
83 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
85 EXPECT_EQ(5, computation->instruction_count()); in TEST_F()
90 EXPECT_EQ(5, computation->instruction_count()); in TEST_F()
148 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
150 EXPECT_EQ(5, computation->instruction_count()); in TEST_F()
[all …]
|
| D | hlo_schedule.h | 121 const HloComputation* computation) const;
126 const HloComputation* computation);
129 void set_sequence(const HloComputation* computation,
131 void set_sequence(const HloComputation* computation,
145 bool is_computation_scheduled(const HloComputation* computation) const { in is_computation_scheduled() argument
146 return sequences_.contains(computation->unique_id()); in is_computation_scheduled()
150 void remove_computation(const HloComputation* computation) { in remove_computation() argument
151 auto it = sequences_.find(computation->unique_id()); in remove_computation()
154 execution_threads_.erase(computation->unique_id()); in remove_computation()
158 void remove_instruction(const HloComputation* computation, in remove_instruction() argument
[all …]
|
| D | hlo_creation_utils.cc | 48 HloComputation* computation = operand->parent(); in MakeUnaryHlo() local
51 return computation->AddInstruction( in MakeUnaryHlo()
63 HloComputation* computation = lhs->parent(); in MakeBinaryHlo() local
64 CHECK_EQ(computation, rhs->parent()); in MakeBinaryHlo()
67 return computation->AddInstruction( in MakeBinaryHlo()
76 HloComputation* computation = lhs->parent(); in MakeCompareHlo() local
77 CHECK_EQ(computation, rhs->parent()); in MakeCompareHlo()
81 return computation->AddInstruction( in MakeCompareHlo()
90 HloComputation* computation = operand->parent(); in MakePadHlo() local
91 CHECK_EQ(computation, padding_value->parent()); in MakePadHlo()
[all …]
|
| D | call_graph.cc | 85 [](std::string* out, const HloComputation* computation) { in ToString() argument
86 out->append(computation->name()); in ToString()
90 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode() argument
91 : computation_(computation) {} in CallGraphNode()
114 CHECK_EQ(instruction->parent(), computation()); in AddCallSiteForInstruction()
136 const HloComputation* computation) const { in GetNode()
137 auto it = node_indices_.find(computation); in GetNode()
142 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode() argument
143 auto it = node_indices_.find(computation); in GetNode()
204 for (const HloComputation* computation : module_->computations()) { in SetCallContexts() local
[all …]
|
| D | loop_schedule_linearizer_test.cc | 39 int64_t 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_t 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().value(); 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()
98 XlaComputation computation = builder.Build().value(); in XLA_TEST_F() local
100 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()
77 CPP_CLASS computation; in BM_NAME() local
78 computation.set_thread_pool(&device); in BM_NAME()
79 zero_buffers(&computation); in BM_NAME()
[all …]
|
| /external/tink/cc/mac/internal/ |
| D | chunked_mac_test.cc | 78 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST_P() local
80 ASSERT_THAT(computation, IsOk()); in TEST_P()
81 ASSERT_THAT((*computation)->Update("abc"), IsOk()); in TEST_P()
82 ASSERT_THAT((*computation)->Update("xyz"), IsOk()); in TEST_P()
83 util::StatusOr<std::string> chunked_tag = (*computation)->ComputeMac(); in TEST_P()
115 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local
117 ASSERT_THAT(computation, IsOk()); in TEST()
118 ASSERT_THAT((*computation)->Update("ab"), IsOk()); in TEST()
119 ASSERT_THAT((*computation)->Update("cx"), IsOk()); in TEST()
120 ASSERT_THAT((*computation)->Update("yz"), IsOk()); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening.cc | 39 HloComputation* computation) { in CreateConstant() argument
44 CreateConstant(shape.tuple_shapes(index), computation); in CreateConstant()
46 return computation->AddInstruction( in CreateConstant()
49 return computation->AddInstruction( in CreateConstant()
73 const HloComputation* computation = while_hlo.parent(); in IsNotContainedInLoop() local
74 while (!computation->IsEntryComputation()) { in IsNotContainedInLoop()
75 auto& node = call_graph.GetNode(computation); in IsNotContainedInLoop()
83 computation = callsite.instruction()->parent(); in IsNotContainedInLoop()
135 HloComputation* computation = while_hlo->parent(); in FlattenWhileLoop() local
137 HloInstruction* initialization = computation->AddInstruction( in FlattenWhileLoop()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 55 HloComputation* computation = operand->parent(); in CreateCholesky() local
96 computation->AddInstruction(HloInstruction::CreateCustomCall( in CreateCholesky()
100 HloInstruction* out = computation->AddInstruction( in CreateCholesky()
102 HloInstruction* info = computation->AddInstruction( in CreateCholesky()
107 HloInstruction* zero = computation->AddInstruction( in CreateCholesky()
110 computation->AddInstruction(HloInstruction::CreateBroadcast( in CreateCholesky()
112 HloInstruction* ok = computation->AddInstruction( in CreateCholesky()
115 ok = computation->AddInstruction(HloInstruction::CreateBroadcast( in CreateCholesky()
121 HloInstruction* nan = computation->AddInstruction( in CreateCholesky()
124 computation->AddInstruction(HloInstruction::CreateBroadcast( in CreateCholesky()
[all …]
|