Searched refs:shape_manager (Results 1 – 1 of 1) sorted by relevance
2140 SymbolicShapeManager* shape_manager) { in ValidateSymbolicShapeManager() argument2155 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager()2166 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager()2189 SymbolicShapeManager* shape_manager) { in VerboseShapeInferenceLogging() argument2218 ic->DebugString(shape_manager->GetMergedShape(ic, ic->input(i))), in VerboseShapeInferenceLogging()2224 ic->DebugString(shape_manager->GetMergedShape(ic, ic->output(i))), in VerboseShapeInferenceLogging()2622 std::unique_ptr<SymbolicShapeManager> shape_manager = in InferStatically() local2636 if (!shape_manager->Merge(merged_shapes.first, merged_shapes.second) in InferStatically()2643 if (!shape_manager->Merge(merged_dims.first, merged_dims.second).ok()) { in InferStatically()2651 shape_manager = std::make_unique<SymbolicShapeManager>(); in InferStatically()[all …]