Searched refs:HloPrintOptions (Results 1 – 22 of 22) sorted by relevance
47 const HloPrintOptions& options) const override;116 const HloPrintOptions& options) const override;144 const HloPrintOptions& options) const override;170 const HloPrintOptions& options) const override;195 const HloPrintOptions& options) const override;228 const HloPrintOptions& options) const override;303 const HloPrintOptions& options) const override;337 const HloPrintOptions& options) const override;385 const HloPrintOptions& options) const override;411 const HloPrintOptions& options) const override;[all …]
67 class HloPrintOptions {77 HloPrintOptions() in HloPrintOptions() function92 static HloPrintOptions ShortParsable() { in ShortParsable()93 return HloPrintOptions() in ShortParsable()106 static HloPrintOptions Canonical() { in Canonical()107 return HloPrintOptions() in Canonical()121 HloPrintOptions& set_print_large_constants(bool value) { in set_print_large_constants()126 HloPrintOptions& set_print_subcomputation_mode( in set_print_subcomputation_mode()133 HloPrintOptions& set_print_metadata(bool value) { in set_print_metadata()139 HloPrintOptions& set_print_backend_config(bool value) { in set_print_backend_config()[all …]
100 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()183 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()221 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()297 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()339 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()379 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()494 const HloPrintOptions& /*options*/) const { in ExtraAttributesToStringImpl()555 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()621 const HloPrintOptions& /*options*/) const { in ExtraAttributesToStringImpl()668 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()[all …]
107 << hlo->ToString(HloPrintOptions().set_print_metadata(false)) in FirstNonScalarAndNonTrivialReshapeOperand()109 << operand->ToString(HloPrintOptions().set_print_metadata(false)); in FirstNonScalarAndNonTrivialReshapeOperand()189 auto print_no_metadata = HloPrintOptions().set_print_metadata(false); in PerformSinkReshapeOrTranspose()265 auto print_no_metadata = HloPrintOptions().set_print_metadata(false); in IsReshapeMoveCandidate()
314 HloPrintOptions().set_indent_amount(1)); in Perform()319 HloPrintOptions().set_indent_amount(1)); in Perform()327 HloPrintOptions().set_indent_amount(1)); in Perform()
180 string ToString() const { return ToString(HloPrintOptions()); } in ToString()181 string ToString(const HloPrintOptions& options) const;185 const HloPrintOptions& options,
202 string ToString() const { return ToString(HloPrintOptions()); } in ToString()203 string ToString(const HloPrintOptions& options) const;
187 module->ToString(HloPrintOptions().set_print_large_constants(false))); in TEST_F()193 module->ToString(HloPrintOptions().set_print_large_constants(true))); in TEST_F()
561 auto options = HloPrintOptions().set_print_metadata(false); in TEST_F()597 HloPrintOptions().set_print_metadata(false).set_indent_amount(2); in TEST_F()632 auto options = HloPrintOptions().set_print_metadata(false); in TEST_F()642 options = HloPrintOptions().Canonical(); in TEST_F()
99 HloPrintOptions().set_compact_operands(true).set_print_operand_names( in CreateHloProfilePrinterData()
128 auto print_no_metadata = HloPrintOptions{}.set_print_metadata(false); in TryHoistingInvariantInstructionsFromWhileBody()
468 string HloComputation::ToString(const HloPrintOptions& options) const { in ToString()473 const HloPrintOptions& options, in ToString()495 HloPrintOptions new_options = options; in ToString()
2106 string PrintName(const string& name, const HloPrintOptions& options) { in PrintName()2112 string HloInstruction::ToString(const HloPrintOptions& options) const { in ToString()2191 const HloPrintOptions& options, in ToStringWithCanonicalNameMap()2231 string HloInstruction::OperandsToString(const HloPrintOptions& options) const { in OperandsToString()2237 const HloPrintOptions& options, in OperandsToStringWithCanonicalNameMap()2276 const HloPrintOptions& options) const { in ExtraAttributesToString()2280 HloPrintOptions::PrintSubcomputationMode::kNameOnly) { in ExtraAttributesToString()2323 HloPrintOptions::PrintSubcomputationMode::kFullBodies) { in ExtraAttributesToString()2324 HloPrintOptions new_options = options; in ExtraAttributesToString()
1328 auto options = HloPrintOptions().set_print_metadata(false); in TEST_F()1493 auto options = HloPrintOptions().Canonical(); in TEST_F()1543 auto options = HloPrintOptions().Canonical(); in TEST_F()1604 auto options = HloPrintOptions().Canonical(); in TEST_F()
1125 HloPrintOptions().set_print_subcomputation_mode( in GetInstructionNodeExtraInfo()1126 HloPrintOptions::PrintSubcomputationMode::kOff))) { in GetInstructionNodeExtraInfo()
200 string HloModule::ToString(const HloPrintOptions& options) const { in ToString()
71 auto print_no_metadata = HloPrintOptions().set_print_metadata(false); in TryRemoveDeadWhileParams()
1144 HloPrintOptions().set_print_metadata(false).set_print_percent(false));
72 module.ToString(HloPrintOptions::ShortParsable()).c_str()); in RealMain()82 module->ToString(HloPrintOptions::ShortParsable()).c_str()); in RealMain()
368 HloPrintOptions::ShortParsable().set_print_backend_config( in DoExtractCommand()
120 ? thunk->hlo_instruction()->ToString(HloPrintOptions::Canonical()) in ExecuteThunks()
382 HloPrintOptions options; in GetHloText()383 options = HloPrintOptions::ShortParsable(); in GetHloText()