Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc27 explicit Visitor(HloComputation* computation) : computation_(computation) {} in Visitor()
45 HloComputation* computation_; member in xla::gpu::__anondf8c4db30111::Visitor
71 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormInference()
74 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormInference()
85 computation_->ReplaceWithNewInstruction(batch_norm, std::move(libcall))); in HandleBatchNormInference()
107 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormTraining()
110 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormTraining()
119 computation_->AddInstruction(HloInstruction::CreateCustomCall( in HandleBatchNormTraining()
127 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining()
130 computation_->AddInstruction(HloInstruction::CreateBinary( in HandleBatchNormTraining()
[all …]
Dfusion_merger.cc133 : computation_(computation) {} in FusionInstructionMerger()
142 HloComputation* computation_; member in xla::gpu::FusionInstructionMerger
159 for (auto* instruction : computation_->MakeInstructionPostOrder()) { in Run()
166 << " computation: " << computation_->name() in Run()
286 return computation_->RemoveInstruction(fusion); in HandleFusion()
Dhlo_execution_profiler.cc62 computation_(computation) { in HloExecutionProfiler()
74 *computation_, in FinishExecution()
Dhlo_execution_profiler.h76 const HloComputation* computation_; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dmap_inliner.cc39 : computation_(computation) {} in MapInlinerVisitor()
53 HloComputation* computation_; member in xla::MapInlinerVisitor
61 computation_ = computation; in Run()
84 TF_RETURN_IF_ERROR(computation_->RemoveInstruction(map)); in HandleMap()
92 HloInstruction* constant = computation_->AddInstruction(root.Clone()); in HandleMap()
93 HloInstruction* placed_instruction = computation_->AddInstruction( in HandleMap()
96 computation_->ReplaceInstruction(map, placed_instruction)); in HandleMap()
102 HloInstruction* placed_instruction = computation_->AddInstruction( in HandleMap()
105 computation_->ReplaceInstruction(map, placed_instruction)); in HandleMap()
Dbfloat16_normalization.cc34 : computation_(computation), bfloat16_support_(bfloat16_support) {} in BFloat16NormalizationVisitor()
74 HloComputation* computation_; member in xla::BFloat16NormalizationVisitor
201 TF_RETURN_IF_ERROR(InsertConvertBeforeOperand(hlo, i, F32, computation_)); in HandleMultipleOutputs()
241 output_elements[i] = computation_->AddInstruction( in HandleMultipleOutputs()
246 auto gte = computation_->AddInstruction( in HandleMultipleOutputs()
249 computation_->AddInstruction(HloInstruction::CreateConvert( in HandleMultipleOutputs()
252 auto tuple = computation_->AddInstruction( in HandleMultipleOutputs()
261 bool is_root = computation_->root_instruction() == hlo; in HandleMultipleOutputs()
263 computation_->set_root_instruction(tuple); in HandleMultipleOutputs()
313 TF_RETURN_IF_ERROR(InsertConvertBeforeOperand(hlo, i, F32, computation_)); in HandleInstruction()
[all …]
Dalgebraic_simplifier.cc258 : computation_(computation), options_(options) {} in AlgebraicSimplifierVisitor()
268 return computation_->AddInstruction(HloInstruction::CreateReshape( in StripDim()
277 return computation_->AddInstruction(HloInstruction::CreateReshape( in Flatten()
290 return computation_->AddInstruction(HloInstruction::CreateConvert( in AsType()
316 computation_->AddInstruction(HloInstruction::CreateConstant( in AddReduce()
322 return computation_->AddInstruction(HloInstruction::CreateReduce( in AddReduce()
372 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in ReplaceWithNewInstruction()
387 computation_->ReplaceInstruction(old_instruction, new_instruction)); in ReplaceInstruction()
413 computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in GetOrCreateScalarAddComputation()
433 HloComputation* computation_; member in xla::__anon4eef97bd0111::AlgebraicSimplifierVisitor
[all …]
Dbatchnorm_expander.cc76 : computation_(computation), in BatchNormExpanderVisitor()
91 return computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in GetOrCreateScalarAddComputation()
142 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in ReplaceWithNewInstruction()
154 computation_->ReplaceInstruction(old_instruction, new_instruction)); in ReplaceInstruction()
160 HloComputation* computation_; member in xla::__anon406107c40111::BatchNormExpanderVisitor
193 HloInstruction* added_inst = computation_->AddInstruction(std::move(inst)); in HandleBatchNormTraining()
202 int64 instruction_count_before = computation_->instruction_count(); in HandleBatchNormTraining()
302 int64 instruction_count_after = computation_->instruction_count(); in HandleBatchNormTraining()
345 auto epsilon = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormInference()
347 computation_->AddInstruction( in HandleBatchNormInference()
[all …]
Dconvolution_group_converter.cc72 : computation_(computation), in ConvolutionVisitor()
78 HloComputation* computation_; member in xla::__anon9c1c05530111::ConvolutionVisitor
215 return computation_->AddInstruction(std::move(inst)); in HandleBatchGroupCount()
281 return computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in HandleBatchGroupCount()
311 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBatchGroupCount()
325 return computation_->AddInstruction(std::move(inst)); in HandleConvolution()
384 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleConvolution()
453 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleConvolution()
531 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleConvolution()
Dbfloat16_conversion_folding.cc33 : computation_(computation), bfloat16_support_(bfloat16_support) {} in BFloat16ConversionFoldingVisitor()
62 HloComputation* computation_; member in xla::BFloat16ConversionFoldingVisitor
169 if (hlo == computation_->root_instruction() && in DefaultAction()
200 if (crs == computation_->root_instruction()) { in HandleAllReduce()
Dinstruction_fusion.cc456 computation_ = computation; in Run()
457 reachability_ = HloReachabilityMap::Build(computation_); in Run()
464 ComputeGloballyUnfusible(computation_->MakeInstructionPostOrder()); in Run()
466 auto fusion_queue = GetFusionQueue(computation_); in Run()
518 TF_RETURN_IF_ERROR(computation_->RemoveInstruction(operand)); in Run()
545 fusion_instruction = computation_->AddInstruction( in AddFusionInstruction()
547 TF_CHECK_OK(computation_->ReplaceInstruction(consumer, fusion_instruction)); in AddFusionInstruction()
Dtuple_points_to_analysis_test.cc829 computation_ = module_->AddEntryComputation(std::move(computation)); in BuildModule()
844 HloComputation* computation_ = nullptr; member in xla::__anon3cbe095a0111::PointsToAnalysisTestBase
896 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
990 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
1051 computation_ = module_->entry_computation(); in TEST_F()
1054 HloInstruction* operand_param = computation_->parameter_instruction(0); in TEST_F()
1055 HloInstruction* indices_param = computation_->parameter_instruction(1); in TEST_F()
1056 HloInstruction* updates_param = computation_->parameter_instruction(2); in TEST_F()
1057 HloInstruction* scatter = computation_->root_instruction(); in TEST_F()
1078 computation_ = module_->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dhlo_memory_scheduler.cc108 : computation_(computation), in ListScheduler()
266 for (auto* instruction : computation_->instructions()) { in CreateSchedule()
291 for (auto* instruction : computation_->instructions()) { in CreateSchedule()
364 CHECK_EQ(schedule.size(), computation_->instruction_count()); in CreateSchedule()
365 CHECK_EQ(scheduled_instructions_.size(), computation_->instruction_count()); in CreateSchedule()
370 HloComputation* computation_; member in xla::__anone86b2dc90111::ListScheduler
Dmulti_output_fusion.h92 HloComputation* computation() const { return computation_; } in computation()
119 HloComputation* computation_; variable
Dhlo_graph_dumper.cc312 : computation_(computation), in HloDotDumper()
379 const HloComputation* computation_; // never null member in xla::__anon7102e5d00111::HloDotDumper
423 StrAppend(&body, DumpComputation(computation_)); in Dump()
460 StrCat(label_, "<br/>Computation ", computation_->name()); in Header()
461 if (computation_->IsFusionComputation()) { in Header()
463 computation_->FusionInstruction()->name(), ")"); in Header()
466 auto cycles = profile_->total_cycles_executed(*computation_); in Header()
682 const HloInstruction* from = GetNodeForEdge(computation_->root_instruction()); in DumpRootTag()
698 auto to_id = SubcomputationId(computation_); in DumpRootTag()
1210 if (instr->parent() != computation_) { in AddInstructionIncomingEdges()
Dmulti_output_fusion.cc31 computation_ = computation; in Run()
38 for (auto it : computation_->MakeInstructionPostOrder()) { in Run()
261 reachability_ = HloReachabilityMap::Build(computation_); in RecomputeReachability()
Dcall_graph.h98 HloComputation* computation() const { return computation_; } in computation()
151 HloComputation* computation_; variable
Dhlo_dataflow_analysis_test.cc1941 computation_ = module_->AddEntryComputation(std::move(computation)); in BuildModule()
1960 HloComputation* computation_ = nullptr; member in xla::__anon497c66660111::HloDataflowAnalysisTestBase
2013 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2056 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2108 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2137 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2168 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2193 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2324 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2367 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
[all …]
Dlayout_assignment.h136 const HloComputation* computation() const { return computation_; } in computation()
137 HloComputation* computation() { return computation_; } in computation()
235 HloComputation* computation_; variable
Dhlo_rematerialization.cc490 const HloComputation* computation_; member in xla::__anon4fe3fcd60111::MemoryUsageTracker
513 : computation_(computation), instruction_list_(instruction_list) { in MemoryUsageTracker()
515 points_to_analysis.GetPointsToSet(computation_->root_instruction()) in MemoryUsageTracker()
751 absl::StrCat("MemoryUsageTracker for ", computation_->name(), "\n"); in ToString()
782 for (auto* instruction : computation_->instructions()) { in Check()
803 for (auto* instruction : computation_->instructions()) { in Check()
Dinstruction_fusion.h119 HloComputation* computation_; variable
Dhlo_computation_test.cc233 : computation_(computation) {} in TEST_F()
243 EXPECT_EQ(computation_->root_instruction(), root); in TEST_F()
248 HloComputation* computation_; in TEST_F() member in xla::__anon457962a40111::TEST_F::TestVisitor
Dlayout_assignment.cc106 : points_to_analysis_(points_to_analysis), computation_(computation) { in LayoutConstraints()
108 for (HloInstruction* inst : computation_->instructions()) { in LayoutConstraints()
273 computation_->name(), curr_shape_layout->ToString(), in SetResultLayout()
357 computation_->name(), ":\n"); in ToString()
358 for (auto* instruction : computation_->MakeInstructionPostOrder()) { in ToString()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduce_window_op.cc34 OP_REQUIRES_OK(context, context->GetAttr("computation", &computation_)); in ReduceWindowOp()
89 compile_options, *computation_, in Compile()
127 const NameAttrList* computation_; member in tensorflow::__anon7be2982c0111::ReduceWindowOp
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc337 : computation_(std::move(computation)) {} in Computation()
354 client.client()->Compile(computation_, argument_shape_pointers, options)); in Compile()
364 const XlaComputation& Computation::computation() const { return computation_; } in computation()
368 if (!computation_.proto().SerializeToString(&result)) { in GetSerializedProto()
378 computation_.proto(), GetDebugOptionsFromFlags())); in GetHloText()
381 HloModule::CreateFromProto(computation_.proto(), module_config)); in GetHloText()
391 computation_.proto(), GetDebugOptionsFromFlags())); in GetHloDotGraph()
394 HloModule::CreateFromProto(computation_.proto(), module_config)); in GetHloDotGraph()
401 return computation_.GetProgramShape(); in GetProgramShape()
405 TF_ASSIGN_OR_RETURN(ProgramShape shape, computation_.GetProgramShape()); in GetReturnValueShape()

12