| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 98 HloPrintOptions options; in GetComputationHloText()
99 options = HloPrintOptions::ShortParsable(); in GetComputationHloText()
292 py::class_<HloPrintOptions> hlo_print_options_class(m, "HloPrintOptions"); in BuildXlaCompilerSubmodule()
294 .def_static("short_parsable", &HloPrintOptions::ShortParsable) in BuildXlaCompilerSubmodule()
295 .def_static("canonical", &HloPrintOptions::Canonical) in BuildXlaCompilerSubmodule()
296 .def_static("fingerprint", &HloPrintOptions::Fingerprint) in BuildXlaCompilerSubmodule()
298 &HloPrintOptions::print_large_constants, in BuildXlaCompilerSubmodule()
299 &HloPrintOptions::set_print_large_constants) in BuildXlaCompilerSubmodule()
300 .def_property("print_metadata", &HloPrintOptions::print_metadata, in BuildXlaCompilerSubmodule()
301 &HloPrintOptions::set_print_metadata) in BuildXlaCompilerSubmodule()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 69 class HloPrintOptions {
79 HloPrintOptions() in HloPrintOptions() function
99 static HloPrintOptions ShortParsable() { in ShortParsable()
100 return HloPrintOptions() in ShortParsable()
113 static HloPrintOptions Canonical() { in Canonical()
114 return HloPrintOptions() in Canonical()
128 static HloPrintOptions Fingerprint() { in Fingerprint()
129 return HloPrintOptions() in Fingerprint()
145 HloPrintOptions& set_print_large_constants(bool value) { in set_print_large_constants()
150 HloPrintOptions& set_print_subcomputation_mode( in set_print_subcomputation_mode()
[all …]
|
| D | hlo_instructions.h | 49 const HloPrintOptions& options) const override;
118 const HloPrintOptions& options) const override;
146 const HloPrintOptions& options) const override;
170 const HloPrintOptions& options) const override;
196 const HloPrintOptions& options) const override;
221 const HloPrintOptions& options) const override;
264 const HloPrintOptions& options) const override;
290 const HloPrintOptions& options) const override;
378 const HloPrintOptions& options) const override;
402 const HloPrintOptions& options) const override;
[all …]
|
| D | hlo_instructions.cc | 103 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
186 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
223 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
269 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
353 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
395 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
440 const HloPrintOptions& /*options*/) const { in ExtraAttributesToStringImpl()
473 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
592 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
628 const HloPrintOptions& options) const { in ExtraAttributesToStringImpl()
[all …]
|
| D | reshape_mover.cc | 108 << hlo->ToString(HloPrintOptions().set_print_metadata(false)) in FirstNonScalarAndNonTrivialReshapeOperand()
110 << operand->ToString(HloPrintOptions().set_print_metadata(false)); in FirstNonScalarAndNonTrivialReshapeOperand()
190 auto print_no_metadata = HloPrintOptions().set_print_metadata(false); in PerformSinkReshapeOrTranspose()
266 auto print_no_metadata = HloPrintOptions().set_print_metadata(false); in IsReshapeMoveCandidate()
|
| D | multi_output_fusion.cc | 423 HloPrintOptions().set_indent_amount(1)); in Perform()
429 HloPrintOptions().set_indent_amount(1)); in Perform()
448 HloPrintOptions().set_indent_amount(1)); in Perform()
|
| D | hlo_module.h | 236 string ToString() const { return ToString(HloPrintOptions()); } in ToString()
237 string ToString(const HloPrintOptions& options) const;
|
| D | hlo_computation.h | 217 string ToString() const { return ToString(HloPrintOptions()); } in ToString()
218 string ToString(const HloPrintOptions& options) const;
222 const HloPrintOptions& options,
|
| D | conditional_code_motion.cc | 547 << conditional_parent->ToString(HloPrintOptions::Fingerprint()) in MoveInstructionOut()
569 << conditional_parent->ToString(HloPrintOptions::Fingerprint()) in MoveInstructionOut()
626 << conditional_parent->ToString(HloPrintOptions::Fingerprint()) in MoveInstructionOut()
641 << conditional->parent()->ToString(HloPrintOptions::Fingerprint()) in MoveInstructionIn()
773 << conditional->ToString(HloPrintOptions::Fingerprint()) << "\n"; in MoveInstructionIn()
802 << conditional->parent()->ToString(HloPrintOptions::Fingerprint()) in MoveInstructionIn()
|
| D | hlo_execution_profile.cc | 100 HloPrintOptions().set_compact_operands(true).set_print_operand_names( in CreateHloProfilePrinterData()
|
| D | hlo_computation_test.cc | 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()
|
| D | hlo_module_test.cc | 188 module->ToString(HloPrintOptions().set_print_large_constants(false))); in TEST_F()
194 module->ToString(HloPrintOptions().set_print_large_constants(true))); in TEST_F()
|
| D | hlo_module.cc | 235 string HloModule::ToString(const HloPrintOptions& options) const { in ToString()
670 return a->ToString(HloPrintOptions::Fingerprint()) < in CompareComputationsByContent()
671 b->ToString(HloPrintOptions::Fingerprint()); in CompareComputationsByContent()
|
| D | while_loop_invariant_code_motion.cc | 128 auto print_no_metadata = HloPrintOptions{}.set_print_metadata(false); in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | while_loop_expensive_invariant_code_motion.cc | 117 auto print_no_metadata = HloPrintOptions{}.set_print_metadata(false); in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | hlo_computation.cc | 539 string HloComputation::ToString(const HloPrintOptions& options) const { in ToString()
544 const HloPrintOptions& options, in ToString()
615 HloPrintOptions new_options = options; in ToString()
|
| D | hlo_instruction_test.cc | 1447 auto options = HloPrintOptions().set_print_metadata(false); in TEST_F()
1453 auto options2 = HloPrintOptions() in TEST_F()
1626 auto options = HloPrintOptions().Canonical(); in TEST_F()
1676 auto options = HloPrintOptions().Canonical(); in TEST_F()
1737 auto options = HloPrintOptions().Canonical(); in TEST_F()
|
| D | hlo_instruction.cc | 2559 string PrintNameInternal(const string& name, const HloPrintOptions& options) { in PrintNameInternal()
2605 string HloInstruction::ToString(const HloPrintOptions& options) const { in ToString()
2689 const HloPrintOptions& options, in ToStringWithCanonicalNameMap()
2738 string HloInstruction::OperandsToString(const HloPrintOptions& options) const { in OperandsToString()
2744 const HloPrintOptions& options, in OperandsToStringWithCanonicalNameMap()
2787 const HloPrintOptions& options) const { in ExtraAttributesToString()
2793 HloPrintOptions::PrintSubcomputationMode::kNameOnly) { in ExtraAttributesToString()
2851 HloPrintOptions::PrintSubcomputationMode::kFullBodies) { in ExtraAttributesToString()
2852 HloPrintOptions new_options = options; in ExtraAttributesToString()
|
| D | while_loop_simplifier.cc | 57 auto print_no_metadata = HloPrintOptions().set_print_metadata(false); in RemoveDeadTupleIndices()
239 auto print_no_metadata = HloPrintOptions().set_print_metadata(false); in TryRemoveDeadWhileParams()
|
| D | hlo_graph_dumper.cc | 1154 HloPrintOptions().set_print_subcomputation_mode( in GetInstructionNodeExtraInfo()
1155 HloPrintOptions::PrintSubcomputationMode::kOff))) { in GetInstructionNodeExtraInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_text.cc | 73 module.ToString(HloPrintOptions::ShortParsable()).c_str()); in RealMain()
83 module->ToString(HloPrintOptions::ShortParsable()).c_str()); in RealMain()
|
| D | interactive_graphviz.cc | 394 HloPrintOptions::ShortParsable().set_print_backend_config( in DoExtractCommand()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_irgen_test_base.cc | 91 HloPrintOptions print_opts; in MatchOptimizedHlo()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | xla_mlir_translate.cc | 200 HloPrintOptions().set_include_layout_in_shapes(with_layouts)); in MlirHloToHloTextTranslateFunctionImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 270 auto options = HloPrintOptions::Canonical(); in AutotuneCacheKeyfromInstruction()
|