Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_proto_util.cc55 const HloProto& hlo_proto) { in EntryComputationParameterShapes() argument
56 if (!hlo_proto.has_hlo_module()) { in EntryComputationParameterShapes()
59 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationParameterShapes()
64 const auto& program_shape = hlo_proto.hlo_module().host_program_shape(); in EntryComputationParameterShapes()
72 const HloProto& hlo_proto) { in EntryComputationOutputShape() argument
73 if (!hlo_proto.has_hlo_module()) { in EntryComputationOutputShape()
76 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationOutputShape()
79 if (!hlo_proto.hlo_module().host_program_shape().has_result()) { in EntryComputationOutputShape()
83 return &hlo_proto.hlo_module().host_program_shape().result(); in EntryComputationOutputShape()
Dhlo_proto_util_test.cc32 HloProto hlo_proto; in TEST_F() local
34 auto status = EntryComputationParameterShapes(hlo_proto).status(); in TEST_F()
41 HloProto hlo_proto; in TEST_F() local
42 HloModuleProto* module = hlo_proto.mutable_hlo_module(); in TEST_F()
45 auto status = EntryComputationParameterShapes(hlo_proto).status(); in TEST_F()
Dhlo_proto_util.h51 const HloProto& hlo_proto);
56 const HloProto& hlo_proto);
Dxla_debug_info_manager.cc148 info.hlo_proto = absl::make_unique<HloProto>( in StopTracing()
150 *info.hlo_proto->mutable_buffer_assignment() = in StopTracing()
Dexecutable.h372 void set_hlo_proto(std::unique_ptr<xla::HloProto> hlo_proto) { in set_hlo_proto() argument
373 hlo_proto_ = std::move(hlo_proto); in set_hlo_proto()
376 HloProto const* hlo_proto() const { return hlo_proto_.get(); } in hlo_proto() function
Dservice.cc295 auto hlo_proto = absl::make_unique<HloProto>(); in BuildExecutables() local
296 *hlo_proto->mutable_hlo_module() = *module_protos[i]; in BuildExecutables()
297 hlo_protos.push_back(std::move(hlo_proto)); in BuildExecutables()
651 *snapshots[i].mutable_hlo() = *executable_ptrs[i]->hlo_proto(); in ExecuteGraphParallel()
782 auto hlo_proto = absl::make_unique<HloProto>(); in BuildExecutable() local
783 *hlo_proto->mutable_hlo_module() = module_proto; in BuildExecutable()
784 executable->set_hlo_proto(std::move(hlo_proto)); in BuildExecutable()
872 *snapshot.mutable_hlo() = *executable->hlo_proto(); in Execute()
Dxla_debug_info_manager.h31 std::unique_ptr<HloProto> hlo_proto; member
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate.cc48 bool LoadHloProto(const std::string& contents, HloProto* hlo_proto) { in LoadHloProto() argument
52 return hlo_proto->ParseFromString(contents) || in LoadHloProto()
53 parser.ParseFromString(contents, hlo_proto) || in LoadHloProto()
54 hlo_proto->mutable_hlo_module()->ParseFromString(contents) || in LoadHloProto()
55 parser.ParseFromString(contents, hlo_proto->mutable_hlo_module()); in LoadHloProto()
63 HloProto hlo_proto; in HloToMlirHloTranslateFunction() local
65 if (!LoadHloProto(content, &hlo_proto)) { in HloToMlirHloTranslateFunction()
73 module.get(), hlo_proto.mutable_hlo_module(), import_all_computations); in HloToMlirHloTranslateFunction()
85 HloProto hlo_proto; in HloTextToMlirHloTranslateFunction() local
124 const HloProto& hlo_proto) { in HloModuleFromProto() argument
[all …]
Dmlir_hlo_to_hlo.h49 Status ConvertMlirHloToHlo(mlir::ModuleOp module, ::xla::HloProto* hlo_proto,
Dmlir_hlo_to_hlo.cc1767 mlir::ModuleOp module, xla::HloProto* hlo_proto, bool use_tuple_args, in ConvertMlirHloToHlo() argument
1782 hlo_proto->mutable_hlo_module()->Swap(&hlo_module); in ConvertMlirHloToHlo()
1784 module, hlo_proto->mutable_hlo_module(), use_tuple_args))) in ConvertMlirHloToHlo()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_proto_to_json.cc60 HloProto hlo_proto; in RealMain() local
62 &hlo_proto)) in RealMain()
65 auto statusor = ToJson(hlo_proto); in RealMain()
Dinteractive_graphviz.cc74 string hlo_proto; member
674 LOG(FATAL) << "Can only specify one and only one of '--hlo_proto', "
748 tensorflow::Flag("hlo_proto", &opts.hlo_proto,
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dmetadata_collector.cc85 xplane.AddStatValue(hlo_proto_stat, *p.hlo_proto); in CollectData()
86 p.hlo_proto.reset(); in CollectData()
/external/tensorflow/tensorflow/core/protobuf/tpu/
DBUILD31 protodeps = ["//tensorflow/compiler/xla/service:hlo_proto"],
69 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtf_xla_mlir_translate.cc279 xla::HloProto hlo_proto; in CompileMlirToXlaHloViaBuilder() local
280 hlo_proto.mutable_hlo_module()->Swap(&hlo_module); in CompileMlirToXlaHloViaBuilder()
284 *xla_computation = xla::XlaComputation(hlo_proto.hlo_module()); in CompileMlirToXlaHloViaBuilder()
Dcompile_mlir_util.cc398 xla::HloProto hlo_proto; in ConvertMLIRToXlaComputation() local
399 TF_RETURN_IF_ERROR(mlir::ConvertMlirHloToHlo(module_op, &hlo_proto, in ConvertMLIRToXlaComputation()
402 *xla_computation = xla::XlaComputation(hlo_proto.hlo_module()); in ConvertMLIRToXlaComputation()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
DBUILD30 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xrt/
DBUILD31 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/core/profiler/
Dprofiler_options.proto51 // Whether serialize hlo_proto when XLA is used. (version >= 1)
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dlibtpu.h118 struct TpuDriver* driver, const struct HloProto hlo_proto,
Dtpu_client.cc837 xla::HloProto hlo_proto; in Compile() local
838 *hlo_proto.mutable_hlo_module() = computation.proto(); in Compile()
843 client->driver()->CompileProgram(hlo_proto, options.num_replicas(), {}); in Compile()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc91 const HloProto& hlo_proto) { in HloModuleFromProto() argument
92 const HloModuleProto& module_proto = hlo_proto.hlo_module(); in HloModuleFromProto()
177 HloProto hlo_proto; in runOnOperation() local
179 ConvertMlirHloToHlo(module, &hlo_proto, in runOnOperation()
185 auto statusOrHloModule = HloModuleFromProto(hlo_proto); in runOnOperation()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc801 xla::HloProto* hlo_proto = proto.add_hlo_protos(); in ComputeInternal() local
802 *hlo_proto = *hlo_metadata; in ComputeInternal()
DBUILD134 "//tensorflow/compiler/xla/service:hlo_proto",
147 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc204 *snapshot->mutable_hlo() = *executable->hlo_proto(); in DumpArguments()

12