Searched refs:hlo_proto (Results 1 – 15 of 15) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_proto_util.cc | 53 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()
|
D | hlo_proto_util_test.cc | 32 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()
|
D | hlo_proto_util.h | 47 const HloProto& hlo_proto); 52 const HloProto& hlo_proto);
|
D | BUILD | 24 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/ |
D | BUILD | 43 "//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",
|
D | hlo_proto_to_json.cc | 60 HloProto hlo_proto; in RealMain() local 62 &hlo_proto)) in RealMain() 65 auto statusor = ToJson(hlo_proto); in RealMain()
|
D | interactive_graphviz.cc | 69 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/ |
D | BUILD | 80 "//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/ |
D | BUILD | 68 "//tensorflow/compiler/xla/service:hlo_proto",
|
/external/tensorflow/tensorflow/compiler/xrt/ |
D | BUILD | 32 "//tensorflow/compiler/xla/service:hlo_proto",
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | BUILD | 102 "//tensorflow/compiler/xla/service:hlo_proto",
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | BUILD | 56 "//tensorflow/compiler/xla/service:hlo_proto",
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | BUILD | 857 "//tensorflow/compiler/xla/service:hlo_proto", 1196 "//tensorflow/compiler/xla/service:hlo_proto",
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | BUILD | 130 "//tensorflow/compiler/xla/service:hlo_proto",
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | BUILD | 1786 "//tensorflow/compiler/xla/service:hlo_proto",
|