| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_rematerialization_test.cc | 60 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
62 HloInstruction::CreateBroadcast(vec1024_shape_, param, {}));
64 HloInstruction::CreateUnary(vec1024_shape_, HloOpcode::kNegate, bcast));
65 auto concat_1 = builder.AddInstruction(HloInstruction::CreateConcatenate(
68 auto slice_1 = builder.AddInstruction(HloInstruction::CreateSlice(
72 auto concat_2 = builder.AddInstruction(HloInstruction::CreateConcatenate(
77 builder.AddInstruction(HloInstruction::CreateSlice(vec1_shape_, concat_2,
103 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
105 HloInstruction::CreateBroadcast(vec1024_shape_, param, {}));
107 HloInstruction::CreateSlice(vec1_shape_, bcast, /*start_indices=*/{0}
[all...] |
| liveness_util.cc | 31 bool DoesNotUseOperandBuffer(const HloInstruction* operand,
33 const HloInstruction* user,
42 user->fusion_kind() == HloInstruction::FusionKind::kLoop) {
46 [=](HloInstruction* fused_param) {
57 for (HloInstruction* alias_user : alias.instruction()->users()) {
72 bool DoesNotUseOperandBuffer(const HloInstruction* operand,
74 const HloInstruction* user,
79 user->fusion_kind() == HloInstruction::FusionKind::kLoop) {
81 HloInstruction* fusion_param =
105 // Each use in 'uses' is a pair (HloInstruction* user, int64 operand_index
[all...] |
| hlo_constant_folding_test.cc | 43 HloInstruction* input = builder.AddInstruction(
44 HloInstruction::CreateConstant(Literal::CreateR0<float>(42.0f)));
46 HloInstruction::CreateConvert(ShapeUtil::MakeShape(S64, {}), input));
64 HloInstruction* input = builder.AddInstruction(
65 HloInstruction::CreateConstant(Literal::CreateR0<int64>(42)));
67 HloInstruction::CreateConvert(ShapeUtil::MakeShape(F32, {}), input));
85 HloInstruction* input = builder.AddInstruction(
86 HloInstruction::CreateConstant(Literal::CreateR1<float>({42.0f, 19.0f})));
88 HloInstruction::CreateConvert(ShapeUtil::MakeShape(S64, {2}), input));
119 std::vector<HloInstruction*> operands
[all...] |
| reshape_mover.cc | 51 bool IsReshapeOrTranspose(const HloInstruction* instruction) {
58 bool CanTriviallyChangeShape(const HloInstruction* instruction) {
97 HloInstruction* FirstNonScalarAndNonTrivialReshapeOperand(
98 const HloInstruction* hlo) {
99 for (HloInstruction* operand : hlo->operands()) {
114 bool AreEquivalentReshapes(const HloInstruction* a, const HloInstruction* b) {
135 const HloInstruction* instruction,
136 const HloInstruction* first_reshape_operand) {
151 for (const HloInstruction* operand : instruction->operands())
[all...] |
| elemental_ir_emitter.h | 34 std::unordered_map<const HloInstruction*, llvm_ir::ElementGenerator>;
44 virtual StatusOr<llvm::Value*> EmitUnaryOp(const HloInstruction* op,
47 virtual StatusOr<llvm::Value*> EmitBinaryOp(const HloInstruction* op,
54 const HloInstruction* hlo,
62 const HloInstruction* op, llvm::Value* operand_value) const;
65 const HloInstruction* op, llvm::Value* operand_value) const;
68 const HloInstruction* op, llvm::Value* operand_value) const;
70 virtual StatusOr<llvm::Value*> EmitIntegerBinaryOp(const HloInstruction* op,
76 const HloInstruction* op, llvm::Value* lhs_value,
80 const HloInstruction* op, llvm::Value* lhs_value
[all...] |
| instruction_fusion.cc | 32 const HloInstruction& instruction) {
134 bool InstructionFusion::EffectivelyUnary(HloInstruction* hlo) {
144 [output_rank](HloInstruction* operand) {
158 const HloReachabilityMap& reachability_map, HloInstruction* producer,
159 HloInstruction* consumer, DoNotFuseSet* do_not_fuse) {
233 // map from HloInstruction* to the instruction's index in the vector. An
236 std::list<HloInstruction*> post_order_list =
238 std::vector<HloInstruction*> post_order(post_order_list.begin(),
241 tensorflow::gtl::FlatMap<HloInstruction*, int> post_order_index;
249 auto cheap_to_duplicate = [this](HloInstruction* producer)
[all...] |
| tuple_points_to_analysis_test.cc | 63 const LogicalBuffer* const GetBuffer(const HloInstruction* instruction,
86 tensorflow::gtl::ArraySlice<HloInstruction*> instructions) {
97 tensorflow::gtl::ArraySlice<HloInstruction*> instructions) {
106 const HloInstruction* instruction, const ShapeIndex& index,
107 tensorflow::gtl::ArraySlice<std::pair<HloInstruction*, ShapeIndex>>
127 HloInstruction::CreateConstant(Literal::CreateR0<float>(1.0)));
129 HloInstruction::CreateConstant(Literal::CreateR0<float>(2.0)));
131 HloInstruction::CreateTuple({constant1, constant2}));
180 HloInstruction::CreateConstant(Literal::CreateR0<float>(1.0)));
182 HloInstruction::CreateConstant(Literal::CreateR0<float>(2.0)))
[all...] |
| buffer_liveness_test.cc | 38 const HloInstruction* instruction,
52 HloInstruction* a, HloInstruction* b) {
64 HloInstruction* a, HloInstruction* b,
89 HloInstruction* instruction) {
96 builder.AddInstruction(HloInstruction::CreateParameter(0, vec_, "param"));
110 builder.AddInstruction(HloInstruction::CreateParameter(0, vec_, "param"));
112 HloInstruction::CreateUnary(vec_, HloOpcode::kNegate, param));
114 HloInstruction::CreateUnary(vec_, HloOpcode::kExp, negate))
[all...] |
| while_util_test.cc | 28 HloComputation** entry_computation, HloInstruction** param0,
29 HloInstruction** param1, HloInstruction** param2) {
63 HloInstruction *param0, *param1, *param2;
70 HloInstruction* while_instr = entry_computation->root_instruction();
77 HloInstruction* new_while_instr = make_live_in_result.new_while_instr;
94 HloInstruction *param0, *param1, *param2;
101 HloInstruction* while_instr = entry_computation->root_instruction();
109 HloInstruction* new_while_instr = make_live_in_result.new_while_instr;
|
| dfs_hlo_visitor.cc | 44 const HloInstruction& instruction) {
50 const HloInstruction& instruction) {
58 const HloInstruction& instruction) {
75 template class DfsHloVisitorBase<HloInstruction*>;
76 template class DfsHloVisitorBase<const HloInstruction*>;
|
| hlo_alias_analysis.h | 64 const HloBuffer& GetUniqueBufferAt(const HloInstruction* instruction,
66 HloBuffer& GetUniqueBufferAt(const HloInstruction* instruction,
73 const HloInstruction* instruction, const ShapeIndex& index = {}) const;
88 bool InstructionBuffersAreAmbiguous(const HloInstruction* instruction) const;
92 bool InstructionBuffersAreDistinct(const HloInstruction* instruction) const;
|
| implicit_broadcast_remover_test.cc | 38 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "p0"));
40 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "p1"));
42 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param0, param1));
56 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
59 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "p1"));
61 HloInstruction::CreateBinary(shape, HloOpcode::kPower, param0, param1));
64 HloInstruction* root = computation->root_instruction();
83 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "p0"));
84 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
86 builder.AddInstruction(HloInstruction::CreateBinary
[all...] |
| hlo_scheduling.cc | 70 static StatusOr<std::vector<const HloInstruction*>> Run(
80 static bool IgnoreInstruction(const HloInstruction& instruction) {
145 // HloInstruction, plus some cached metadata, saved for the purposes of making
148 const HloInstruction* instruction;
162 ReadyListEntry MakeReadyListEntry(const HloInstruction* instruction) {
204 std::vector<const HloInstruction*> CreateSchedule() {
205 std::vector<const HloInstruction*> schedule;
209 tensorflow::gtl::FlatMap<const HloInstruction*, int64>
213 // predecessors() when these methods are added to HloInstruction.
214 for (const HloInstruction* user : instruction->users())
[all...] |
| hlo_tfgraph_builder_test.cc | 37 HloInstruction::CreateParameter(0, r0f32_, "param0"));
39 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, param));
48 HloInstruction::CreateParameter(0, r0f32_, "param0"));
50 HloInstruction::CreateMap(r0f32_, {param}, map_computation));
67 HloInstruction::CreateParameter(0, shape, "param0"));
69 HloInstruction::CreateParameter(1, shape, "param1"));
70 builder.AddInstruction(HloInstruction::CreateConcatenate(
93 HloInstruction *instruction = builder.AddInstruction(
94 HloInstruction::CreateConstant(Literal::CreateR0(123)));
125 HloInstruction::CreateParameter(0, r0f32_, "param0"))
[all...] |
| bfloat16_normalization.h | 75 bool SupportsBF16Operand(const HloInstruction& hlo,
80 bool SupportsBF16Output(const HloInstruction& hlo) const override {
84 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override {
|
| call_graph.h | 56 CallContext GetInstructionCallContext(const HloInstruction* instruction);
61 CallSite(HloInstruction* instruction,
69 HloInstruction* instruction() const { return instruction_; }
83 HloInstruction* instruction_;
108 const CallSite* GetCallSite(const HloInstruction* instruction) const;
140 void AddCallSiteForInstruction(HloInstruction* instruction);
160 tensorflow::gtl::FlatMap<const HloInstruction*, int64> callsite_instructions_;
201 bool InstructionIsNestedIn(const HloInstruction* instruction,
229 std::pair<HloInstruction*, HloInstruction*> NearestAncestorsInSameComputation
[all...] |
| hlo_evaluator_test.cc | 74 b.AddInstruction(HloInstruction::CreateConstant(std::move(input)));
76 HloInstruction::CreateUnary(expected->shape(), opcode, c1));
94 auto c1 = b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs)));
95 auto c2 = b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs)));
97 HloInstruction::CreateBinary(expected->shape(), opcode, c1, c2));
120 auto c1 = b.AddInstruction(HloInstruction::CreateConstant(std::move(low)));
121 auto c2 = b.AddInstruction(HloInstruction::CreateConstant(std::move(value)));
122 auto c3 = b.AddInstruction(HloInstruction::CreateConstant(std::move(high)));
124 HloInstruction::CreateTernary(shape, HloOpcode::kClamp, c1, c2, c3));
141 auto c1 = b.AddInstruction(HloInstruction::CreateConstant(std::move(low)))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| hlo_schedule.cc | 38 const std::vector<const HloInstruction*>& thunk_launch_order);
43 const std::vector<const HloInstruction*>* SequentialOrder(
52 std::unique_ptr<std::vector<const HloInstruction*>> entry_sequence_;
57 const std::vector<const HloInstruction*>& thunk_launch_order)
62 MakeUnique<std::vector<const HloInstruction*>>(thunk_launch_order);
81 std::vector<const HloInstruction*> last_instruction_per_stream(
84 for (const HloInstruction* hlo : thunk_launch_order) {
89 std::vector<const HloInstruction*> immediate_preds;
150 std::vector<const HloInstruction*>* launch_order) {
158 std::deque<const HloInstruction*> queue
[all...] |
| pad_insertion.cc | 29 bool IsForwardConvolutionCanonical(const HloInstruction& conv) {
42 HloInstruction* MaybePaddedAndSlicedInput(
44 HloInstruction* input) {
68 HloInstruction* padding =
69 computation->AddInstruction(HloInstruction::CreateConstant(
71 input = computation->AddInstruction(HloInstruction::CreatePad(
100 input = computation->AddInstruction(HloInstruction::CreateSlice(
114 HloInstruction* MaybePaddedKernel(const Window& conv_window,
116 HloInstruction* kernel) {
134 HloInstruction* padding
[all...] |
| gpu_layout_assignment_test.cc | 54 HloInstruction::CreateParameter(0, ashape, "x"));
56 HloInstruction::CreateParameter(1, ashape, "y"));
58 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, x, y));
75 for (const HloInstruction* operand : add->operands()) {
123 HloInstruction::CreateParameter(0, shape, "operand"));
125 HloInstruction::CreateParameter(1, aux_shape, "scale"));
127 HloInstruction::CreateParameter(2, aux_shape, "offset"));
129 HloInstruction::CreateParameter(3, aux_shape, "mean"));
131 HloInstruction::CreateParameter(4, aux_shape, "variance"));
134 HloInstruction::CreateConstant(Literal::CreateR0<float>(1)))
[all...] |
| while_transformer_test.cc | 45 HloInstruction::CreateConstant(Literal::CreateR0<int32>(limit)));
46 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter(
49 builder.AddInstruction(HloInstruction::CreateGetTupleElement(
52 HloInstruction::CreateBinary(condition_result_shape_, HloOpcode::kLt,
62 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter(
66 builder.AddInstruction(HloInstruction::CreateGetTupleElement(
69 HloInstruction::CreateConstant(Literal::CreateR0<int32>(increment)));
70 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary(
73 auto data = builder.AddInstruction(HloInstruction::CreateGetTupleElement(
77 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {}))
[all...] |
| thunk.h | 33 // metadata IrEmitter generates for GpuExecutable to invoke an HloInstruction.
62 explicit Thunk(Kind kind, const HloInstruction* hlo_instruction)
69 const HloInstruction* hlo_instruction() const { return hlo_instruction_; }
110 const HloInstruction* hlo_instruction_;
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| broadcast_test.cc | 40 HloInstruction::CreateConstant(Literal::CreateR0<float>(42.0)));
41 builder.AddInstruction(HloInstruction::CreateBroadcast(
56 HloInstruction::CreateConstant(Literal::CreateR0<float>(42.0)));
57 builder.AddInstruction(HloInstruction::CreateBroadcast(
72 auto input = builder.AddInstruction(HloInstruction::CreateConstant(
77 auto element1 = builder.AddInstruction(HloInstruction::CreateBroadcast(
79 auto element2 = builder.AddInstruction(HloInstruction::CreateBroadcast(
81 builder.AddInstruction(HloInstruction::CreateTuple({element1, element2}));
99 auto input = builder.AddInstruction(HloInstruction::CreateConstant(
101 builder.AddInstruction(HloInstruction::CreateBroadcast
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_copy_insertion_test.cc | 71 HloInstruction::CreateParameter(0, scalar_shape_, "param_0"));
75 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
77 HloInstruction::CreateConstant(Literal::CreateR0<float>(123.0)));
82 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
84 HloInstruction::CreateConstant(Literal::CreateR0<bool>(false)));
89 HloInstruction::CreateWhile(scalar_shape_, condition, body, param_0));
109 HloInstruction::CreateParameter(0, scalar_shape_, "param_0"));
115 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
117 HloInstruction::CreateConstant(Literal::CreateR0<float>(123.0)));
118 auto add = sub_builder.AddInstruction(HloInstruction::CreateBinary
[all...] |
| parallel_task_assignment.h | 30 virtual int64 GetParallelTaskCount(HloInstruction* instruction) = 0;
46 int64 GetTargetParallelTaskCount(HloInstruction* instruction);
79 using HloToParallelTasks = std::unordered_map<const HloInstruction*, int64>;
|
