Searched refs:op_supports_dynamism_handler_ (Results 1 – 2 of 2) sorted by relevance
70 op_supports_dynamism_handler_(op_supports_dynamism_handler) {} in slice_dynamic_output_()86 OpSupportsDynamismHandler op_supports_dynamism_handler_; variable
1563 : op_supports_dynamism_handler_(op_supports_dynamism_handler), in DynamicShapeRemovingVisitor()1604 const DynamicPadder::OpSupportsDynamismHandler& op_supports_dynamism_handler_; member in xla::__anon6d34db3b0111::DynamicShapeRemovingVisitor1692 if (op_supports_dynamism_handler_) { in DefaultAction()1693 op_support = op_supports_dynamism_handler_(hlo); in DefaultAction()1815 if (op_supports_dynamism_handler_ != nullptr) { in Run()1816 has_dynamism_support = op_supports_dynamism_handler_(inst); in Run()1937 computation, op_supports_dynamism_handler_, dynamic_dimension_inference, in Run()