| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_alias_analysis.cc | 64 : module_(module), dataflow_(dataflow) {
65 buffers_.reserve(dataflow_.values().size());
66 value_to_buffer_number_.reserve(dataflow_.values().size());
67 for (const HloValue* value : dataflow_.values()) {
77 for (const HloValue* value : dataflow_.values()) {
181 return dataflow_.GetUniqueValueAt(
218 dataflow_.GetUniqueValueAt(use.instruction, use.operand_index);
230 dataflow_.call_graph().GetNode(computation);
236 const HloValue& while_value = dataflow_.GetUniqueValueAt(
249 dataflow_.call_graph().GetNode(computation)
331 const HloDataflowAnalysis& dataflow_; member in class:xla::BufferValueMap
[all...] |
| bfloat16_propagation.h | 207 std::unique_ptr<HloDataflowAnalysis> dataflow_; member in class:xla::BFloat16Propagation
|
| bfloat16_propagation.cc | 91 dataflow_->GetValueSet(root, entry.second).values()) {
112 dataflow_->GetValueSet(inst, index).values()) {
214 auto& value_set = dataflow_->GetValueSet(&hlo, index);
346 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) {
472 for (const auto* value : dataflow_->GetValueSet(root, index).values()) {
518 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) {
534 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) {
754 TF_ASSIGN_OR_RETURN(dataflow_, HloDataflowAnalysis::Run(*module));
[all...] |
| copy_insertion.cc | 477 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) {
496 dataflow_) ||
498 dataflow_))
571 dataflow_.GetValueSet(instruction->operand(0));
575 value_to_node.at(&dataflow_.GetUniqueValueAt(instruction));
940 const HloDataflowAnalysis& dataflow_; member in namespace:xla::__anon44320
[all...] |
