Searched refs:ShapeComputeGraphMapping (Results 1 – 4 of 4) sorted by relevance
20 struct ShapeComputeGraphMapping { struct21 ShapeComputeGraphMapping( in ShapeComputeGraphMapping() function38 TORCH_API std::optional<ShapeComputeGraphMapping> argument
23 const ShapeComputeGraphMapping& shape_mapping, in InsertSymbolicShapesCompute()73 const ShapeComputeGraphMapping& shape_mapping,341 const ShapeComputeGraphMapping& shape_mapping, in insertDynamicShapesGuard()
814 std::optional<ShapeComputeGraphMapping> run() { in run()910 return ShapeComputeGraphMapping( in run()1121 std::optional<ShapeComputeGraphMapping>
1141 py::class_<torch::jit::ShapeComputeGraphMapping>( in initPythonIRBindings()1145 [](ShapeComputeGraphMapping& g) { in initPythonIRBindings()1150 [](ShapeComputeGraphMapping& g) { in initPythonIRBindings()