Searched refs:DynamicDimensionInference (Results 1 – 8 of 8) sorted by relevance
30 DynamicDimensionInference* parent) in DynamicDimensionInferenceVisitor()37 DynamicDimensionInference* parent) { in Run()95 DynamicDimensionInference* parent_;575 void DynamicDimensionInference::CopyMapping(HloInstruction* from, in CopyMapping()590 StatusOr<DynamicDimensionInference> DynamicDimensionInference::Run( in Run()593 DynamicDimensionInference inference(module); in Run()598 string DynamicDimensionInference::ToString() const { in ToString()612 DynamicDimensionInference::DynamicDimensionInference(HloModule* module) in DynamicDimensionInference() function in xla::DynamicDimensionInference615 Status DynamicDimensionInference::AnalyzeDynamicDimensions() { in AnalyzeDynamicDimensions()620 HloInstruction* DynamicDimensionInference::GetDynamicSize( in GetDynamicSize()
31 const DynamicDimensionInference* dynamic_dimension_inference) { in ReplaceGetSize()62 TF_ASSIGN_OR_RETURN(DynamicDimensionInference inference, in Run()63 DynamicDimensionInference::Run(module)); in Run()
37 class DynamicDimensionInference {39 static StatusOr<DynamicDimensionInference> Run(HloModule* module);52 explicit DynamicDimensionInference(HloModule* module);
111 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in Run()112 DynamicDimensionInference::Run(module)); in Run()
130 DynamicDimensionInference* dynamic_dimension_inference) { in set_dynamic_dimension_inference()343 DynamicDimensionInference* dynamic_dimension_inference_ = nullptr;
46 TF_ASSIGN_OR_RETURN(DynamicDimensionInference inference, in RunInference()47 DynamicDimensionInference::Run(module_.get())); in RunInference()49 inference_ = absl::make_unique<DynamicDimensionInference>(inference); in RunInference()76 std::unique_ptr<DynamicDimensionInference> inference_;
1100 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in ComputeConstantGraph()1101 DynamicDimensionInference::Run(module.get())); in ComputeConstantGraph()
3050 TF_ASSERT_OK_AND_ASSIGN(DynamicDimensionInference dynamic_dimension_inference, in TEST_F()3051 DynamicDimensionInference::Run(m_.get())); in TEST_F()