/external/tensorflow/tensorflow/core/profiler/convert/ |
D | hlo_to_tools_data.cc | 42 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
|
D | hlo_proto_to_graph_view.cc | 195 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()
|
D | hlo_proto_to_memory_visualization_utils.cc | 258 const HloProto& hlo_proto, int64_t small_buffer_size, in ConvertHloProtoToPreprocessResult()
|
/external/tensorflow/tensorflow/core/profiler/utils/ |
D | hlo_proto_map.cc | 61 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()
|
D | hlo_proto_to_module.cc | 31 const xla::HloProto& hlo_proto) { in ConvertHloProtoToModule() 44 const xla::HloProto& hlo_proto) { in ConvertHloProtoToModuleIgnoringErrors()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_proto_util_test.cc | 32 HloProto hlo_proto; in TEST_F() local 41 HloProto hlo_proto; in TEST_F() local
|
D | hlo_proto_util.cc | 57 const HloProto& hlo_proto) { in EntryComputationParameterShapes() 74 const HloProto& hlo_proto) { in EntryComputationOutputShape()
|
D | xla_debug_info_manager.cc | 88 auto hlo_proto = std::make_unique<HloProto>(MakeHloProto(*m.hlo_module)); in StopTracing() local
|
D | executable.h | 370 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
|
D | xla_debug_info_manager_test.cc | 46 for (const auto& hlo_proto : module_debug_info) { in StopTracing() local
|
/external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
D | jax_to_tfl_flatbuffer.cc | 66 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/ |
D | xla_mlir_translate.cc | 42 bool LoadHloProto(const std::string& contents, HloProto* hlo_proto) { in LoadHloProto() 57 HloProto hlo_proto; in HloToMlirHloTranslateFunction() local
|
D | xla_mlir_translate_registration.cc | 94 const HloProto& hlo_proto) { in HloModuleFromProto() 105 ::xla::HloProto* hlo_proto, in ConvertMlirHloToHloViaBuilder()
|
D | mlir_hlo_to_hlo.cc | 2563 Status ConvertMlirHloToHlo(mlir::ModuleOp module, xla::HloProto* hlo_proto, in ConvertMlirHloToHlo()
|
/external/tensorflow/tensorflow/core/profiler/backends/cpu/ |
D | metadata_utils.h | 35 void AddHloProto(uint64_t program_id, const xla::HloProto& hlo_proto) { in AddHloProto()
|
D | metadata_collector.cc | 65 for (auto& hlo_proto : debug_info_) { in CollectData() local
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | hlo_proto_to_json.cc | 61 HloProto hlo_proto; in RealMain() local
|
D | interactive_graphviz.cc | 79 std::string hlo_proto; member
|
/external/tensorflow/tensorflow/core/tpu/kernels/ |
D | tpu_compile_op_common.cc | 403 xla::HloProto* hlo_proto = proto.add_hlo_protos(); in ComputeInternal() local
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
D | tf_xla_mlir_translate.cc | 293 xla::HloProto hlo_proto; in CompileMlirToXlaHloViaBuilder() local
|
D | compile_mlir_util.cc | 546 xla::HloProto hlo_proto; in ConvertMLIRToXlaComputation() local
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | mhlo_to_lhlo_with_xla.cc | 98 const HloProto& hlo_proto) { in HloModuleFromProto() 201 HloProto hlo_proto; in runOnOperation() local
|
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
D | tpu_client.cc | 847 xla::HloProto hlo_proto; in Compile() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_compiler.cc | 1283 auto hlo_proto = std::make_unique<HloProto>(); in CompileLegacyCpuExecutable() local
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_compiler.cc | 1426 auto hlo_proto = std::make_unique<HloProto>(); in RunBackend() local
|