Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc29 explicit Visitor(HloComputation* computation) : computation_(computation) {} in Visitor()
47 HloComputation* computation_; member in xla::gpu::__anonbbe443b50111::Visitor
50 return this->computation_->AddInstruction( in AddConvert()
86 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormInference()
89 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormInference()
109 computation_->AddInstruction(std::move(batchnorm_inference_result)); in HandleBatchNormInference()
115 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBatchNormInference()
138 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormTraining()
141 computation_->AddInstruction(HloInstruction::CreateConstant( in HandleBatchNormTraining()
164 computation_->AddInstruction(HloInstruction::CreateCustomCall( in HandleBatchNormTraining()
[all …]
Dfusion_merger.cc133 : computation_(computation) {} in FusionInstructionMerger()
142 HloComputation* computation_; member in xla::gpu::FusionInstructionMerger
160 for (auto* instruction : computation_->MakeInstructionPostOrder()) { in Run()
167 << " computation: " << computation_->name() in Run()
292 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.cc41 : computation_(nullptr), in BFloat16NormalizationVisitor()
89 HloComputation* computation_; member in xla::__anonddaad1370111::BFloat16NormalizationVisitor
313 InsertConvertBeforeOperand(hlo, i, BF16, F32, computation_)); in HandleMultipleOutputs()
353 output_elements[i] = computation_->AddInstruction( in HandleMultipleOutputs()
359 auto gte = computation_->AddInstruction( in HandleMultipleOutputs()
364 computation_->AddInstruction(HloInstruction::CreateConvert(shape, gte)); in HandleMultipleOutputs()
366 auto tuple = computation_->AddInstruction( in HandleMultipleOutputs()
375 bool is_root = computation_->root_instruction() == hlo; in HandleMultipleOutputs()
377 computation_->set_root_instruction(tuple); in HandleMultipleOutputs()
428 InsertConvertBeforeOperand(hlo, i, BF16, F32, computation_)); in HandleInstruction()
[all …]
Dalgebraic_simplifier.cc388 return computation_->AddInstruction( in AsType()
417 HloInstruction* zero = computation_->AddInstruction( in AddReduce()
426 return computation_->AddInstruction(HloInstruction::CreateReduce( in AddReduce()
488 computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in GetOrCreateScalarAddComputation()
511 HloComputation* computation_; member in xla::__anon2e7aabbf0111::AlgebraicSimplifierVisitor
530 computation_ = computation; in ResetState()
561 auto bitcast = computation_->AddInstruction( in ReplaceWithBitcast()
626 sum_of_constants = computation_->AddInstruction( in HandleAdd()
660 computation_->AddInstruction(HloInstruction::CreateDynamicSlice( in HandleAdd()
664 auto add_partial = computation_->AddInstruction( in HandleAdd()
[all …]
Dbatchnorm_expander.cc68 : computation_(computation), in BatchNormExpanderVisitor()
83 return computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in GetOrCreateScalarAddComputation()
130 HloComputation* computation_; member in xla::__anon42b333460111::BatchNormExpanderVisitor
160 HloInstruction* added_inst = computation_->AddInstruction(std::move(inst)); in HandleBatchNormTraining()
169 int64 instruction_count_before = computation_->instruction_count(); in HandleBatchNormTraining()
269 int64 instruction_count_after = computation_->instruction_count(); in HandleBatchNormTraining()
312 auto epsilon = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormInference()
314 computation_->AddInstruction( in HandleBatchNormInference()
328 HloInstruction* added_inst = computation_->AddInstruction(std::move(inst)); in HandleBatchNormInference()
342 int64 instruction_count_before = computation_->instruction_count(); in HandleBatchNormInference()
[all …]
Dconvolution_group_converter.cc72 : computation_(computation), in ConvolutionVisitor()
78 HloComputation* computation_; member in xla::__anon2e6c43150111::ConvolutionVisitor
214 return computation_->AddInstruction(std::move(inst)); in HandleBatchGroupCount()
304 TF_CHECK_OK(computation_->ReplaceInstruction( in HandleBatchGroupCount()
373 return computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in HandleBatchGroupCount()
403 TF_CHECK_OK(computation_->ReplaceWithNewInstruction( in HandleBatchGroupCount()
417 return computation_->AddInstruction(std::move(inst)); in HandleConvolution()
479 return computation_->ReplaceWithNewInstruction( in HandleConvolution()
509 computation_->AddInstruction(HloInstruction::CreateReshape( in HandleConvolution()
523 computation_->AddInstruction(HloInstruction::CreateReshape( in HandleConvolution()
[all …]
Ddepthwise_convolution_converter.cc65 : computation_(computation), is_cost_viable_(is_cost_viable) {} in ConvolutionVisitor()
68 HloComputation* computation_; member in xla::__anon447517f50111::ConvolutionVisitor
136 return computation_->AddInstruction(std::move(inst)); in HandleBackwardFilterBatchGroupConvolution()
175 auto new_convolution = computation_->AddInstruction(std::move(new_conv)); in HandleBackwardFilterBatchGroupConvolution()
184 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBackwardFilterBatchGroupConvolution()
Dbfloat16_conversion_folding.cc35 : computation_(computation), in BFloat16ConversionFoldingVisitor()
68 HloComputation* computation_; member in xla::BFloat16ConversionFoldingVisitor
178 if (hlo == computation_->root_instruction() && in DefaultAction()
209 if (crs == computation_->root_instruction()) { in HandleAllReduce()
Dinstruction_fusion.cc483 computation_ = computation; in Run()
484 reachability_ = HloReachabilityMap::Build(computation_); in Run()
491 ComputeGloballyUnfusible(computation_->MakeInstructionPostOrder()); in Run()
493 auto fusion_queue = GetFusionQueue(computation_); in Run()
562 TF_RETURN_IF_ERROR(computation_->RemoveInstruction(operand)); in Run()
610 fusion_instruction = computation_->AddInstruction( in AddFusionInstruction()
612 TF_CHECK_OK(computation_->ReplaceInstruction(consumer, fusion_instruction)); in AddFusionInstruction()
Dmulti_output_fusion.h98 HloComputation* computation() const { return computation_; } in computation()
184 HloComputation* computation_; variable
Dhlo_memory_scheduler.cc110 : 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::__anonc7f641cb0111::ListScheduler
Dcall_graph.h98 HloComputation* computation() const { return computation_; } in computation()
151 HloComputation* computation_; variable
Dhlo_graph_dumper.cc316 : computation_(computation), in HloDotDumper()
383 const HloComputation* computation_; // never null member in xla::__anon627048d20111::HloDotDumper
427 StrAppend(&body, DumpComputation(computation_)); in Dump()
464 StrCat(label_, "<br/>Computation ", computation_->name()); in Header()
465 if (computation_->IsFusionComputation()) { in Header()
467 computation_->FusionInstruction()->name(), ")"); in Header()
470 auto cycles = profile_->total_cycles_executed(*computation_); in Header()
686 const HloInstruction* from = GetNodeForEdge(computation_->root_instruction()); in DumpRootTag()
704 auto to_id = SubcomputationId(computation_); in DumpRootTag()
1229 if (instr->parent() != computation_) { in AddInstructionIncomingEdges()
Dhlo_dataflow_analysis_test.cc2015 computation_ = module_->AddEntryComputation(std::move(computation)); in BuildModule()
2033 HloComputation* computation_ = nullptr; member in xla::__anon8bfaf3e80111::HloDataflowAnalysisTestBase
2086 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2129 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2181 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2210 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2241 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2266 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2397 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2440 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
[all …]
Dlayout_assignment.h137 const HloComputation* computation() const { return computation_; } in computation()
138 HloComputation* computation() { return computation_; } in computation()
236 HloComputation* computation_; variable
Dmulti_output_fusion.cc33 computation_ = computation; in Run()
40 for (auto it : computation_->MakeInstructionPostOrder()) { in Run()
342 reachability_ = HloReachabilityMap::Build(computation_); in RecomputeReachability()
Dhlo_computation.h110 : computation_(computation), metadata_(metadata) {} in MetadataBuilder()
115 return computation_->AddInstruction(std::move(instruction)); in AddInstruction()
119 HloComputation* computation_;
Dinstruction_fusion.h132 HloComputation* computation_; variable
Dhlo_rematerialization.cc591 const HloComputation* computation_; member in xla::__anon20d191180111::MemoryUsageTracker
627 : computation_(computation), in MemoryUsageTracker()
633 points_to_analysis.GetPointsToSet(computation_->root_instruction()) in MemoryUsageTracker()
1007 absl::StrCat("MemoryUsageTracker for ", computation_->name(), "\n"); in ToString()
1049 for (auto* instruction : computation_->instructions()) { in Check()
1070 for (auto* instruction : computation_->instructions()) { in Check()
Dcall_graph.cc86 : computation_(computation) {} in CallGraphNode()
97 std::string CallGraphNode::ToString() const { return computation_->name(); } in ToString()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduce_window_op.cc34 OP_REQUIRES_OK(context, context->GetAttr("computation", &computation_)); in ReduceWindowOp()
88 compile_options, *computation_, in Compile()
126 const NameAttrList* computation_; member in tensorflow::__anonc87a33ee0111::ReduceWindowOp
/external/perfetto/src/trace_processor/sqlite/
Ddb_sqlite_table.cc91 computation_(context.computation), in DbSqliteTable()
154 switch (computation_) { in BestIndex()
424 switch (db_sqlite_table_->computation_) { in Filter()

12