Searched refs:hlo_query (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_query.h | 27 namespace hlo_query {
|
| D | hlo_constant_folding.cc | 85 if (!hlo_query::AllOperandsAreConstants(*instruction)) { in Run()
|
| D | map_inliner.cc | 71 if (hlo_query::AllOperandsAreParameters(root)) { in HandleMap()
|
| D | gather_expander_test.cc | 202 ASSERT_FALSE(hlo_query::ContainsInstrWithOpcode(module->entry_computation(), in TEST_F()
|
| D | hlo_query.cc | 25 namespace hlo_query { namespace
|
| D | BUILD | 293 ":hlo_query",
1635 name = "hlo_query",
1636 srcs = ["hlo_query.cc"],
1637 hdrs = ["hlo_query.h"],
2036 ":hlo_query",
2135 ":hlo_query",
2201 ":hlo_query",
2315 ":hlo_query",
2531 ":hlo_query",
4100 ":hlo_query",
[all …]
|
| D | all_reduce_combiner.cc | 276 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()
|
| D | while_loop_all_reduce_code_motion.cc | 433 hlo_query::NextChannelId(*(while_parent->parent())), in CreateSinkedAllReduces()
|
| D | ar_crs_combiner.cc | 52 int64 next_channel = hlo_query::NextChannelId(*module); in ReplaceReplicatedAllReduce()
|
| D | while_loop_simplifier.cc | 38 using hlo_query::ContainsInstrWithOpcode;
|
| D | hlo_evaluator.cc | 301 if (!hlo_query::AllOperandsAreConstants(*instruction)) { in Evaluate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | BUILD | 51 "//tensorflow/compiler/xla/service:hlo_query",
113 "//tensorflow/compiler/xla/service:hlo_query",
|
| D | canonicalize_all_gather_for_cse.cc | 99 next_channel_id_ = hlo_query::NextChannelId(*module); in Run()
|
| D | spmd_partitioner.cc | 3674 int64 next_channel_id = hlo_query::NextChannelId(*module); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 935 "//tensorflow/compiler/xla/service:hlo_query",
|