| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_query.h | 29 bool IsConstantR0F32(HloInstruction* instruction, float* out);
33 bool AllOperandsAreParametersOrConstants(const HloInstruction& instruction);
36 bool AllOperandsAreParameters(const HloInstruction& instruction);
39 bool AllOperandsAreConstants(const HloInstruction& instruction);
42 bool IsScalarConstant(const HloInstruction* instruction);
47 HloInstruction* GetMatchingOperand(
48 std::function<bool(const HloInstruction*)> matcher,
49 HloInstruction* instruction);
56 std::function<bool(const HloInstruction*)> matcher,
57 HloInstruction* instruction, HloInstruction** matching_operand
[all...] |
| tuple_util.h | 30 static HloInstruction* ExtractPrefix(HloInstruction* input_tuple,
39 static HloInstruction* AppendSuffix(
40 HloInstruction* input_tuple,
41 tensorflow::gtl::ArraySlice<HloInstruction*> trailing_values);
|
| bfloat16_support.h | 31 virtual bool SupportsBF16Operand(const HloInstruction& hlo,
35 virtual bool SupportsBF16Output(const HloInstruction& hlo) const;
40 virtual bool SupportsMixedPrecisions(const HloInstruction& hlo) const;
50 const HloInstruction& hlo, int64 operand_index);
54 virtual bool EffectiveOperandPrecisionIsBF16(const HloInstruction& hlo,
|
| liveness_util.h | 36 bool DoesNotUseOperandBuffer(const HloInstruction* operand,
38 const HloInstruction* user,
40 bool DoesNotUseOperandBuffer(const HloInstruction* operand,
42 const HloInstruction* user,
53 HloInstruction* operand, const ShapeIndex& operand_index,
54 HloInstruction* user, const ShapeIndex& user_index,
56 bool CanShareOperandBufferWithUser(HloInstruction* operand,
58 HloInstruction* user,
|
| hlo_query.cc | 25 bool IsConstantR0F32(HloInstruction* instruction, float* out) {
35 bool AllOperandsAreParametersOrConstants(const HloInstruction& instruction) {
45 bool AllOperandsAreParameters(const HloInstruction& instruction) {
54 bool AllOperandsAreConstants(const HloInstruction& instruction) {
63 HloInstruction* GetMatchingOperand(
64 std::function<bool(const HloInstruction*)> matcher,
65 HloInstruction* instruction) {
66 for (HloInstruction* op : instruction->operands()) {
75 std::function<bool(const HloInstruction*)> matcher,
76 HloInstruction* instruction, HloInstruction** matching_operand
[all...] |
| instruction_fusion.h | 35 std::function<bool(const HloInstruction& instruction)> is_expensive,
48 static bool IsExpensive(const HloInstruction& instruction);
62 virtual bool ShouldFuse(HloInstruction* consumer, int64 operand_index);
66 virtual HloInstruction::FusionKind ChooseKind(const HloInstruction* producer,
67 const HloInstruction* consumer);
70 virtual HloInstruction* Fuse(HloInstruction* producer,
71 HloInstruction* consumer);
77 bool EffectivelyUnary(HloInstruction* hlo)
[all...] |
| while_util.h | 28 HloInstruction* new_while_instr;
32 std::vector<HloInstruction*> while_body_live_in_values;
53 HloInstruction* while_instr,
54 tensorflow::gtl::ArraySlice<HloInstruction*> instructions);
|
| bfloat16_conversion_folding_test.cc | 35 bool SupportsBF16Operand(const HloInstruction& hlo,
46 bool SupportsBF16Output(const HloInstruction& hlo) const override {
56 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override {
81 HloInstruction* a = builder.AddInstruction(
82 HloInstruction::CreateParameter(0, f32_shape, "a"));
83 HloInstruction* b = builder.AddInstruction(
84 HloInstruction::CreateParameter(1, f32_shape, "b"));
85 HloInstruction* c = builder.AddInstruction(
86 HloInstruction::CreateParameter(2, f32_shape, "c"));
88 HloInstruction* add0 = builder.AddInstruction
[all...] |
| hlo_computation.h | 61 HloInstruction* fusion_instruction = nullptr)
71 HloInstruction* root_instruction = nullptr);
73 HloInstruction* AddInstruction(
74 std::unique_ptr<HloInstruction> instruction) {
81 const std::function<Status(const HloInstruction*)>& func) const {
90 HloInstruction* last_added_instruction_;
91 HloInstruction* fusion_instruction_;
92 std::vector<std::unique_ptr<HloInstruction>> instructions_;
97 HloInstruction* AddInstruction(std::unique_ptr<HloInstruction> instruction)
[all...] |
| hlo_dataflow_analysis.h | 68 bool ValueIsDefinedAt(const HloInstruction* instruction,
75 const HloValue& GetValueDefinedAt(const HloInstruction* instruction,
77 HloValue& GetValueDefinedAt(const HloInstruction* instruction,
82 const HloInstruction* instruction) const;
84 const HloInstruction* instruction);
88 const HloValueSet& GetValueSet(const HloInstruction* instruction,
92 HloValueSet& GetValueSet(const HloInstruction* instruction,
97 const HloValue& GetUniqueValueAt(const HloInstruction* instruction,
101 HloValue& GetUniqueValueAt(const HloInstruction* instruction,
126 HloValue* NewHloValue(HloInstruction* instruction, const ShapeIndex& index
[all...] |
| hlo_cost_analysis.h | 52 Status HandleElementwiseUnary(const HloInstruction* hlo) override;
53 Status HandleElementwiseBinary(const HloInstruction* hlo) override;
54 Status HandleConstant(const HloInstruction* constant) override;
56 const HloInstruction* get_tuple_element) override;
57 Status HandleSelect(const HloInstruction* select) override;
58 Status HandleCompare(const HloInstruction* compare) override;
59 Status HandleClamp(const HloInstruction* clamp) override;
60 Status HandleReducePrecision(const HloInstruction* hlo) override;
61 Status HandleConcatenate(const HloInstruction* concatenate) override;
62 Status HandleSend(const HloInstruction* send) override
[all...] |
| hlo_verifier.h | 34 Status HandleElementwiseUnary(HloInstruction* hlo) override;
35 Status HandleElementwiseBinary(HloInstruction* hlo) override;
36 Status HandleClamp(HloInstruction* clamp) override;
37 Status HandleSelect(HloInstruction* select) override;
38 Status HandleConcatenate(HloInstruction* concatenate) override;
39 Status HandleConvert(HloInstruction* convert) override;
40 Status HandleBitcastConvert(HloInstruction* convert) override;
41 Status HandleCopy(HloInstruction* copy) override;
42 Status HandleDot(HloInstruction* dot) override;
43 Status HandleConvolution(HloInstruction* convolution) override
[all...] |
| hlo_scheduling_test.cc | 43 HloInstruction* cond_param = cond_builder.AddInstruction(
44 HloInstruction::CreateParameter(0, tuple_shape, "cond_param"));
45 HloInstruction* cond_iter = cond_builder.AddInstruction(
46 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0));
47 HloInstruction* cond_data = cond_builder.AddInstruction(
48 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1));
50 HloInstruction* cond_lt = cond_builder.AddInstruction(
51 HloInstruction::CreateBinary(ShapeUtil::MakeShape(PRED, {}),
58 HloInstruction* body_param = body_builder.AddInstruction(
59 HloInstruction::CreateParameter(0, tuple_shape, "body_param"))
[all...] |
| hlo_instruction.h | 162 class HloInstruction {
175 ~HloInstruction();
182 // instruction_map: a map from instruction name to HloInstruction*. This map
190 static StatusOr<std::unique_ptr<HloInstruction>> CreateFromProto(
192 const tensorflow::gtl::FlatMap<string, HloInstruction*>& instruction_map,
198 static std::unique_ptr<HloInstruction> CreateParameter(int64 parameter_number,
203 static std::unique_ptr<HloInstruction> CreateConstant(
207 static std::unique_ptr<HloInstruction> CreateGetTupleElement(
208 const Shape& shape, HloInstruction* operand, int64 index);
211 static std::unique_ptr<HloInstruction> CreateTrace(const string& tag
[all...] |
| tuple_simplifier_test.cc | 55 HloInstruction* param0 = builder.AddInstruction(
56 HloInstruction::CreateParameter(0, scalar_shape_, "param0"));
57 HloInstruction* param1 = builder.AddInstruction(
58 HloInstruction::CreateParameter(1, scalar_shape_, "param1"));
59 HloInstruction* param2 = builder.AddInstruction(
60 HloInstruction::CreateParameter(2, scalar_shape_, "param2"));
61 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1, param2}));
71 HloInstruction* param = builder.AddInstruction(
72 HloInstruction::CreateParameter(0, tuple_shape_, "param"));
74 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1))
[all...] |
| call_inliner.h | 31 std::unordered_map<HloInstruction*, HloInstruction*>;
35 static StatusOr<InlinedInstructionMap> Inline(HloInstruction* call);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion.h | 32 bool ShouldFuse(HloInstruction* consumer, int64 operand_index) override;
33 HloInstruction::FusionKind ChooseKind(
34 const HloInstruction* producer, const HloInstruction* consumer) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_copy_insertion.h | 37 StatusOr<HloInstruction*> FindOrInsertCopy(HloInstruction* hlo);
41 tensorflow::gtl::FlatMap<HloInstruction*, HloInstruction*> hlo_to_copy_map_;
|
| instruction_fusion.h | 30 bool ShouldFuse(HloInstruction* consumer, int64 operand_index) override;
32 HloInstruction::FusionKind ChooseKind(
33 const HloInstruction* producer, const HloInstruction* consumer) override;
|
| stream_assignment.h | 26 // This class encapsulates the assignment of GPU streams to each HloInstruction.
30 int StreamNumberForHlo(const HloInstruction& hlo) const;
31 bool HasStreamAssigned(const HloInstruction& hlo) const;
33 void AssignStreamToHlo(const HloInstruction* hlo, int stream_no);
37 tensorflow::gtl::FlatMap<const HloInstruction*, int> hlo_to_stream_number_;
|
| stream_assignment_test.cc | 37 HloInstruction* x = builder.AddInstruction(HloInstruction::CreateParameter(
39 HloInstruction* y = builder.AddInstruction(HloInstruction::CreateParameter(
41 HloInstruction* z = builder.AddInstruction(HloInstruction::CreateParameter(
43 HloInstruction* dot1 = builder.AddInstruction(
44 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, x, y));
45 HloInstruction* dot2 = builder.AddInstruction(
46 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, dot1, z))
[all...] |
| ir_emitter_unnested.h | 28 // or more kernels for each HloInstruction it contains. Examples of unnested
59 Status DefaultAction(HloInstruction* hlo) override;
63 Status HandleCopy(HloInstruction* copy) override;
64 Status HandleConditional(HloInstruction* conditional) override;
65 Status HandleConvolution(HloInstruction* convolution) override;
66 Status HandleCustomCall(HloInstruction* custom_call) override;
67 Status HandleDot(HloInstruction* dot) override;
68 Status HandleFft(HloInstruction* fft) override;
69 Status HandleFusion(HloInstruction* fusion) override;
70 Status HandleGather(HloInstruction* gather) override
[all...] |
| ir_emission_utils.h | 34 bool ImplementedAsGemm(const HloInstruction& hlo);
59 bool IsCustomCallToDnnBatchNorm(const HloInstruction& hlo);
97 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo);
106 HloInstruction* CreateCudnnConvForward(
107 const Shape& shape, HloInstruction* input, HloInstruction* kernel,
109 HloInstruction* CreateCudnnConvBackwardInput(
110 const Shape& shape, HloInstruction* output, HloInstruction* reverse_filter,
112 HloInstruction* CreateCudnnConvBackwardFilter
[all...] |
| ir_emitter.h | 72 Status DefaultAction(HloInstruction* hlo) override;
73 Status HandleConstant(HloInstruction* constant) override;
74 Status HandleBitcast(HloInstruction* bitcast) override;
75 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
76 Status HandleDot(HloInstruction* dot) override;
77 Status HandleConvolution(HloInstruction* convolution) override;
78 Status HandleFft(HloInstruction* fft) override;
79 Status HandleCrossReplicaSum(HloInstruction* crs) override;
80 Status HandleInfeed(HloInstruction* infeed) override;
81 Status HandleOutfeed(HloInstruction* outfeed) override
[all...] |
| pad_insertion.h | 35 bool CanonicalizeForwardConvolution(HloInstruction* conv);
36 bool CanonicalizeBackwardFilterConvolution(HloInstruction* backward_conv);
37 bool CanonicalizeBackwardInputConvolution(HloInstruction* backward_conv);
|
