/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);
|