Home
last modified time | relevance | path

Searched defs:hlo_proto (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/convert/
Dhlo_to_tools_data.cc42 const xla::HloProto& hlo_proto) { in ConvertHloProtoToMemoryViewer()
71 const xla::HloProto& hlo_proto, const ToolOptions& options) { in ConvertHloProtoToGraphViewer()
105 xla::HloProto hlo_proto; in ConvertHloProtoToToolData() local
Dhlo_proto_to_graph_view.cc195 const HloProto& hlo_proto, const std::string& node_name, int graph_width, in ConvertHloProtoToGraph()
203 StatusOr<std::string> ConvertHloProtoToStringView(const HloProto& hlo_proto, in ConvertHloProtoToStringView()
Dhlo_proto_to_memory_visualization_utils.cc258 const HloProto& hlo_proto, int64_t small_buffer_size, in ConvertHloProtoToPreprocessResult()
/external/tensorflow/tensorflow/core/profiler/utils/
Dhlo_proto_map.cc61 auto hlo_proto = std::make_unique<xla::HloProto>(); in ParseHloProtosFromXSpace() local
82 auto hlo_proto = std::make_unique<xla::HloProto>(); in ParseHloProtosFromXSpace() local
97 const xla::HloProto* hlo_proto) { in AddHloProto()
110 std::unique_ptr<const xla::HloProto> hlo_proto) { in AddHloProto()
Dhlo_proto_to_module.cc31 const xla::HloProto& hlo_proto) { in ConvertHloProtoToModule()
44 const xla::HloProto& hlo_proto) { in ConvertHloProtoToModuleIgnoringErrors()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_proto_util_test.cc32 HloProto hlo_proto; in TEST_F() local
41 HloProto hlo_proto; in TEST_F() local
Dhlo_proto_util.cc57 const HloProto& hlo_proto) { in EntryComputationParameterShapes()
74 const HloProto& hlo_proto) { in EntryComputationOutputShape()
Dxla_debug_info_manager.cc88 auto hlo_proto = std::make_unique<HloProto>(MakeHloProto(*m.hlo_module)); in StopTracing() local
Dexecutable.h370 void set_hlo_proto(std::unique_ptr<xla::HloProto> hlo_proto) { in set_hlo_proto()
376 HloProto const* hlo_proto() const { return hlo_proto_.get(); } in hlo_proto() function
Dxla_debug_info_manager_test.cc46 for (const auto& hlo_proto : module_debug_info) { in StopTracing() local
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Djax_to_tfl_flatbuffer.cc66 bool LoadHloProto(const std::string& contents, xla::HloProto* hlo_proto) { in LoadHloProto()
79 xla::HloProto hlo_proto; in HloToMlirHloTranslateFunction() local
101 xla::HloProto hlo_proto; in HloTextToMlirHloTranslateFunction() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate.cc42 bool LoadHloProto(const std::string& contents, HloProto* hlo_proto) { in LoadHloProto()
57 HloProto hlo_proto; in HloToMlirHloTranslateFunction() local
Dxla_mlir_translate_registration.cc94 const HloProto& hlo_proto) { in HloModuleFromProto()
105 ::xla::HloProto* hlo_proto, in ConvertMlirHloToHloViaBuilder()
Dmlir_hlo_to_hlo.cc2563 Status ConvertMlirHloToHlo(mlir::ModuleOp module, xla::HloProto* hlo_proto, in ConvertMlirHloToHlo()
/external/tensorflow/tensorflow/core/profiler/backends/cpu/
Dmetadata_utils.h35 void AddHloProto(uint64_t program_id, const xla::HloProto& hlo_proto) { in AddHloProto()
Dmetadata_collector.cc65 for (auto& hlo_proto : debug_info_) { in CollectData() local
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_proto_to_json.cc61 HloProto hlo_proto; in RealMain() local
Dinteractive_graphviz.cc79 std::string hlo_proto; member
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc403 xla::HloProto* hlo_proto = proto.add_hlo_protos(); in ComputeInternal() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtf_xla_mlir_translate.cc293 xla::HloProto hlo_proto; in CompileMlirToXlaHloViaBuilder() local
Dcompile_mlir_util.cc546 xla::HloProto hlo_proto; in ConvertMLIRToXlaComputation() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc98 const HloProto& hlo_proto) { in HloModuleFromProto()
201 HloProto hlo_proto; in runOnOperation() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc847 xla::HloProto hlo_proto; in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc1283 auto hlo_proto = std::make_unique<HloProto>(); in CompileLegacyCpuExecutable() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc1426 auto hlo_proto = std::make_unique<HloProto>(); in RunBackend() local