HomeSort by relevance Sort by last modified time
    Searched refs:HloDataflowAnalysis (Results 1 - 15 of 15) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
copy_insertion.h 55 CopyInsertion(const HloDataflowAnalysis::FusionCanShareBufferFunction&
107 HloDataflowAnalysis::FusionCanShareBufferFunction fusion_can_share_buffer_;
hlo_alias_analysis.h 45 const HloDataflowAnalysis::FusionCanShareBufferFunction&
85 const HloDataflowAnalysis& dataflow_analysis() const {
111 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis_;
hlo_dataflow_analysis.cc 44 HloDataflowAnalysis::HloDataflowAnalysis(
53 bool HloDataflowAnalysis::AreTransitiveUsesElementwiseOrTuple(
78 bool HloDataflowAnalysis::ValueIsDefinedAt(const HloInstruction* instruction,
87 const HloValue& HloDataflowAnalysis::GetValueDefinedAt(
93 HloValue& HloDataflowAnalysis::GetValueDefinedAt(
99 HloValue* HloDataflowAnalysis::NewHloValue(HloInstruction* instruction,
113 void HloDataflowAnalysis::MarkValueForDeletion(HloValue::Id value_id) {
120 void HloDataflowAnalysis::DeleteMarkedValues() {
147 string HloDataflowAnalysis::ToString() const
    [all...]
hlo_ordering.h 55 const HloDataflowAnalysis& dataflow) const;
59 const HloDataflowAnalysis& dataflow) const;
64 const HloDataflowAnalysis& dataflow) const;
hlo_dataflow_analysis.h 43 class HloDataflowAnalysis {
76 static StatusOr<std::unique_ptr<HloDataflowAnalysis>> Run(
157 HloDataflowAnalysis(
bfloat16_propagation.h 207 std::unique_ptr<HloDataflowAnalysis> dataflow_;
hlo_ordering.cc 172 const HloDataflowAnalysis& dataflow) const {
257 const HloDataflowAnalysis& dataflow) const {
298 const HloDataflowAnalysis& dataflow) const {
hlo_ordering_test.cc 188 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true));
240 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true));
373 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true));
429 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true));
485 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true));
hlo_dataflow_analysis_test.cc 51 const HloDataflowAnalysis& RunAnalysis(bool ssa_form,
54 HloDataflowAnalysis::Run(*module_, ssa_form, bitcast_defines_value)
93 std::unique_ptr<HloDataflowAnalysis> analysis_;
114 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form);
163 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form);
219 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form);
275 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form);
329 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form);
382 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form);
445 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form)
    [all...]
hlo_alias_analysis.cc 63 const HloDataflowAnalysis& dataflow)
331 const HloDataflowAnalysis& dataflow_;
491 HloModule* module, const HloDataflowAnalysis::FusionCanShareBufferFunction&
498 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true,
copy_insertion.cc 168 bool IndicesToCopyForWhile(const HloDataflowAnalysis& dataflow,
    [all...]
bfloat16_propagation.cc 754 TF_ASSIGN_OR_RETURN(dataflow_, HloDataflowAnalysis::Run(*module));
    [all...]
buffer_liveness_test.cc 690 auto dataflow = HloDataflowAnalysis::Run(*module).ConsumeValueOrDie();
    [all...]
tuple_points_to_analysis.cc 733 return HloDataflowAnalysis::AreTransitiveUsesElementwiseOrTuple(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
test_utils.cc 393 const HloDataflowAnalysis& dataflow, const HloInstruction& param) {
533 StatusOr<Literal> MakeConstrainedArgument(const HloDataflowAnalysis& dataflow,
557 TF_ASSIGN_OR_RETURN(auto dataflow, HloDataflowAnalysis::Run(*module));

Completed in 912 milliseconds