Searched refs:dynamic_dimension_inference_ (Results 1 – 5 of 5) sorted by relevance
47 dynamic_dimension_inference_(std::move(dynamic_dymension_inference)) { in InterpreterExecutable()48 if (dynamic_dimension_inference_.has_value()) { in InterpreterExecutable()50 &dynamic_dimension_inference_.value()); in InterpreterExecutable()
64 absl::optional<DynamicDimensionInference> dynamic_dimension_inference_;
133 dynamic_dimension_inference_ = dynamic_dimension_inference; in set_dynamic_dimension_inference()137 return dynamic_dimension_inference_; in dynamic_dimension_inference()367 DynamicDimensionInference* dynamic_dimension_inference_ = nullptr; variable
1564 dynamic_dimension_inference_(dynamic_dimension_inference) {} in DynamicShapeRemovingVisitor()1606 const DynamicDimensionInference& dynamic_dimension_inference_; member in xla::__anon6d34db3b0111::DynamicShapeRemovingVisitor1617 if (dynamic_dimension_inference_.HasDynamicDimension(operand)) { in ConvertToDynamic()1634 dynamic_dimension_inference_.GetDynamicSize(inst, {}, i); in ConvertToDynamic()1731 if (dynamic_dimension_inference_.HasDynamicDimension(operand)) { in DefaultAction()
416 if (dynamic_dimension_inference_ == nullptr) { in HandleGetDimensionSize()422 dynamic_dimension_inference_->GetDynamicSize(operand, {}, dim); in HandleGetDimensionSize()1930 dynamic_dimension_inference_); in HandleCall()1965 dynamic_dimension_inference_); in HandleFusion()1990 dynamic_dimension_inference_); in HandleConditional()2039 cond_evaluator.set_dynamic_dimension_inference(dynamic_dimension_inference_); in HandleWhile()2042 dynamic_dimension_inference_); in HandleWhile()