Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_proto_util.cc53 const HloProto& hlo_proto) { in EntryComputationParameterShapes() argument
54 if (!hlo_proto.has_hlo_module()) { in EntryComputationParameterShapes()
57 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationParameterShapes()
62 const auto& program_shape = hlo_proto.hlo_module().host_program_shape(); in EntryComputationParameterShapes()
70 const HloProto& hlo_proto) { in EntryComputationOutputShape() argument
71 if (!hlo_proto.has_hlo_module()) { in EntryComputationOutputShape()
74 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationOutputShape()
77 if (!hlo_proto.hlo_module().host_program_shape().has_result()) { in EntryComputationOutputShape()
81 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.h47 const HloProto& hlo_proto);
52 const HloProto& hlo_proto);
DBUILD24 name = "hlo_proto",
31 name = "hlo_proto", # bzl adds a _py suffix only to the OSS target.
351 ":hlo_proto",
908 ":hlo_proto",
1125 ":hlo_proto",
1182 ":hlo_proto",
1225 ":hlo_proto",
1262 ":hlo_proto",
1277 ":hlo_proto",
2297 ":hlo_proto",
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
DBUILD43 "//tensorflow/compiler/xla/service:hlo_proto",
69 "//tensorflow/compiler/xla/service:hlo_proto",
121 "//tensorflow/compiler/xla/service:hlo_proto",
151 "//tensorflow/compiler/xla/service:hlo_proto",
171 "//tensorflow/compiler/xla/service:hlo_proto",
186 "//tensorflow/compiler/xla/service:hlo_proto",
228 "//tensorflow/compiler/xla/service:hlo_proto",
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.cc69 string hlo_proto; member
677 } else if (!opts.hlo_proto.empty()) { in RealMain()
679 opts.hlo_proto, xla::GetDebugOptionsFromFlags()) in RealMain()
721 tensorflow::Flag("hlo_proto", &opts.hlo_proto, in main()
/external/tensorflow/tensorflow/compiler/xla/client/
DBUILD80 "//tensorflow/compiler/xla/service:hlo_proto",
123 "//tensorflow/compiler/xla/service:hlo_proto",
202 "//tensorflow/compiler/xla/service:hlo_proto",
227 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/core/protobuf/tpu/
DBUILD68 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xrt/
DBUILD32 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xla/python/
DBUILD102 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xla/
DBUILD56 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD857 "//tensorflow/compiler/xla/service:hlo_proto",
1196 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD130 "//tensorflow/compiler/xla/service:hlo_proto",
/external/tensorflow/tensorflow/compiler/xla/tests/
DBUILD1786 "//tensorflow/compiler/xla/service:hlo_proto",